MaroneAI commited on
Commit
8a723a2
·
verified ·
1 Parent(s): 5406833

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,10 +1,10 @@
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
- # Charger le modèle depuis ton repo Hugging Face
5
  model_name = "MaroneAI/Niani-Translator_French-To-Wolof"
6
 
7
- # ✅ Correction : ignorer le tokenizer.json et forcer SentencePiece
8
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
9
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
10
 
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
+ # Charger le modèle depuis Hugging Face
5
  model_name = "MaroneAI/Niani-Translator_French-To-Wolof"
6
 
7
+ # ✅ Correction : ignorer le tokenizer.json et forcr SentencePiece
8
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
9
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
10