Update app.py
Browse files
app.py
CHANGED
|
@@ -44,7 +44,7 @@ def english_to_indian(sentence):
|
|
| 44 |
generated_tokens = translation_model.generate(**encoded_hi,
|
| 45 |
forced_bos_token_id=translation_tokenizer.lang_code_to_id[lang_global] )
|
| 46 |
x = translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
|
| 47 |
-
translated_sentence = translated_sentence + x
|
| 48 |
return translated_sentence
|
| 49 |
|
| 50 |
def indian_to_english(sentence):
|
|
@@ -55,7 +55,7 @@ def indian_to_english(sentence):
|
|
| 55 |
encoded_hi = translation_tokenizer(chunk, return_tensors="pt")
|
| 56 |
generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id["en_XX"] )
|
| 57 |
x = translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
|
| 58 |
-
translated_sentence = translated_sentence + x
|
| 59 |
return translated_sentence
|
| 60 |
|
| 61 |
|
|
|
|
| 44 |
generated_tokens = translation_model.generate(**encoded_hi,
|
| 45 |
forced_bos_token_id=translation_tokenizer.lang_code_to_id[lang_global] )
|
| 46 |
x = translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
|
| 47 |
+
translated_sentence = translated_sentence + x[0]
|
| 48 |
return translated_sentence
|
| 49 |
|
| 50 |
def indian_to_english(sentence):
|
|
|
|
| 55 |
encoded_hi = translation_tokenizer(chunk, return_tensors="pt")
|
| 56 |
generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id["en_XX"] )
|
| 57 |
x = translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
|
| 58 |
+
translated_sentence = translated_sentence + x[0]
|
| 59 |
return translated_sentence
|
| 60 |
|
| 61 |
|