You can deploy this model and start earning money today!

drixo/
multilingual-doc-assistant

Name: multilingual-doc-assistant
Availability: InStock

text-to-speech

Views

No views yet

Multilingual Document Assistant

Agent-style model for explaining documents, answering questions, and responding conversationally in:

Spanish
Chinese
Vietnamese
Portuguese

Base model: bigscience/bloom-560m on Hugging Face.

Run on Hugging Face

To run this as a Hugging Face Space (browser chat UI):

Create a Space at huggingface.co/new-space:
- Choose Gradio.
- Clone or upload this repo (at least app.py and requirements.txt).
Use your fine-tuned model (after training and pushing):
- Train: python train.py
- Push to Hub: export HF_REPO_ID=your-username/multilingual-doc-assistant then python push_to_hub.py
- In the Space, go to Settings → Variables and add:
  - HF_MODEL_ID = your-username/multilingual-doc-assistant
- The app will load your model from the Hub. Without this, it uses the base BLOOM model.
The Space runs app.py and serves the Gradio chat interface.

Setup (local)

bash

cd multilingual-doc-assistant
pip install -r requirements.txt

Train

python train.py

Saves the fine-tuned model and tokenizer to ./multilingual-doc-model. You can run from any directory; paths are relative to the script.

Test / Chat

After training:

python test_model.py

Uses a Spanish prompt by default. You can edit the prompt in test_model.py to try other languages or questions.

Training data

Add more examples in train.jsonl (one JSON object per line with a "text" key). Use the same User: / Assistant: format so the model learns the conversational style.

Run the Space UI locally