app90 commited on
Commit
074931c
·
verified ·
1 Parent(s): b0da48c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,7 +18,8 @@ except RuntimeError:
18
  # ✅ Load Fine-Tuned Chhattisgarhi Translation Model
19
  model_path = "app90/ChhattishgarhiAI_Model"
20
  model = MBartForConditionalGeneration.from_pretrained(model_path)
21
- tokenizer = MBart50TokenizerFast.from_pretrained(model_path, src_lang="hi_IN", tgt_lang="hne_IN")
 
22
  translator = Translator()
23
 
24
  # ✅ Detect Language
 
18
  # ✅ Load Fine-Tuned Chhattisgarhi Translation Model
19
  model_path = "app90/ChhattishgarhiAI_Model"
20
  model = MBartForConditionalGeneration.from_pretrained(model_path)
21
+ # tokenizer = MBart50TokenizerFast.from_pretrained(model_path, src_lang="hi_IN", tgt_lang="hne_IN")
22
+ tokenizer = MBart50Tokenizer.from_pretrained(model_path, src_lang="hi_IN", tgt_lang="hne_IN")
23
  translator = Translator()
24
 
25
  # ✅ Detect Language