Update app.py

app.py

@@ -20,7 +20,7 @@ martian_mt_tokenizer = MarianTokenizer.from_pretrained("AbhirupGhosh/opus-mt-fin
 
 def translate_english_to_hindi(english_text):
     tokenized_text = martian_mt_tokenizer.encode(english_text, return_tensors="pt")
-    generated_token_ids = martian_mt_model.generate(tokenized_text, use_cache=True)
+    generated_token_ids = martian_mt_model.generate(tokenized_text, use_cache=True, max_new_tokens=128)
     hindi_text = martian_mt_tokenizer.decode(generated_token_ids.numpy()[0])
     hindi_text = hindi_text.replace("</s>", "")
     hindi_text = hindi_text.replace("<pad>", "")
@@ -29,12 +29,12 @@ def translate_english_to_hindi(english_text):
 
 
 def translate_to_english(audio):
-    outputs = asr_pipe(audio, generate_kwargs={"task": "transcribe", "use_cache":"True"})
+    outputs = asr_pipe(audio, generate_kwargs={"task": "transcribe", "use_cache":"True", "max_new_tokens":"128"})
     return outputs["text"]
 
 def synthesise(text):
     inputs = processor(text=text, return_tensors="pt").to(device)
-    speech_values = model.generate(**inputs, use_cache=True)
+    speech_values = model.generate(**inputs, use_cache=True, max_new_tokens=128)
     speech_values = speech_values.cpu().numpy()
 
     return speech_values