Update app.py

app.py

@@ -1,7 +1,10 @@
 import streamlit as st
-from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoTokenizer, AutoModelForSeq2SeqLM, pipeline, logging
 import torch
 import os
+import httpx
+
+logging.set_verbosity_error()
 
 # Create the app layout
 st.header("Text Machine Translation")
@@ -54,8 +57,7 @@ if submit_button:
         output_ids = model.generate(input_ids)
         # Decode the translated text
         translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
-    elif model_name.startswith('Google'):
-        import httpx
+    elif model_name.startswith('Google'): 
         url = os.environ['GCLIENT'] + f'sl={sl}&tl={tl}&q={input_text}'
         response = httpx.get(url)
         translated_text = response.json()[0][0][0]
@@ -79,6 +81,7 @@ if submit_button:
         start_marker = "<end_of_turn>"
         if start_marker in translated_text:
             translated_text = translated_text.split(start_marker)[1].strip()
+        translated_text = translated_text.replace('Answer:', '').strip() if translated_text.startswith('Answer:') else translated_text 
         
     # Display the translated text
     print(translated_text)