Update app.py

app.py

@@ -37,6 +37,10 @@ def translate_speech(audio_file_path):
     if 'generated_token_ids' in translated_text[0]:
         # Decode the tokens into text
         translated_text_str = translator.tokenizer.decode(translated_text[0]['generated_token_ids'])
+        
+        # Remove special tokens
+        translated_text_str = translated_text_str.replace("<pad>", "").replace("</s>", "").strip()
+
         print(f"Translated text string: {translated_text_str}")  # Print the translated text string
     else:
         print("The translated text does not contain 'generated_token_ids'")