Spaces:
Runtime error
Runtime error
shubhankargaur
commited on
Commit
•
c689e33
1
Parent(s):
d0daec4
Update app.py
Browse files
app.py
CHANGED
@@ -8,10 +8,10 @@ tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-one-to
|
|
8 |
|
9 |
def get_input(text):
|
10 |
models_input = tokenizer(text,return_tensors="pt")
|
11 |
-
generated_tokens = model.generate(**models_input,forced_bos_token_id=tokenizer.lang_code_to_id["
|
12 |
translation = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
|
13 |
return translation
|
14 |
|
15 |
-
iface = gr.Interface(fn=get_input,inputs="text",outputs="text", title = "English to
|
16 |
|
17 |
iface.launch()
|
|
|
8 |
|
9 |
def get_input(text):
|
10 |
models_input = tokenizer(text,return_tensors="pt")
|
11 |
+
generated_tokens = model.generate(**models_input,forced_bos_token_id=tokenizer.lang_code_to_id["hi_IN"])
|
12 |
translation = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
|
13 |
return translation
|
14 |
|
15 |
+
iface = gr.Interface(fn=get_input,inputs="text",outputs="text", title = "English to Hindi Translator",description="Translate the text given in English to Hindi")
|
16 |
|
17 |
iface.launch()
|