h4d35 commited on
Commit
3c02171
1 Parent(s): e5ce979

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
- model_translate = AutoModelForSeq2SeqLM.from_pretrained("facebook/m2m100_418M")
5
- tokenizer_translate = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
6
  inlang='hi'
7
  outlang='en'
8
  tokenizer_translate.src_lang = inlang
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
+ model_translate = AutoModelForSeq2SeqLM.from_pretrained("h4d35/Translator")
5
+ tokenizer_translate = AutoTokenizer.from_pretrained("h4d35/Translator")
6
  inlang='hi'
7
  outlang='en'
8
  tokenizer_translate.src_lang = inlang