Update app.py

app.py

@@ -8,7 +8,7 @@ input_text = st.text_input("Enter text to translate:")
 # Create a list of options for the select box
 options = ["German", "Romanian", "English", "French", "Spanish"]
 langs = {"English":"en", "Romanian":"ro", "German":"de", "French":"fr", "Spanish":"es", "Italian":"it"}
-models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large", "Unbabel/TowerInstruct-7B-v0.2", "Unbabel/TowerInstruct-Mistral-7B-v0.2"]
+models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large", "Unbabel/TowerInstruct-7B-v0.2", "Unbabel/TowerInstruct-Mistral-7B-v0.2", 'Google']
 
 # Create two columns
 scol, tcol = st.columns(2)
@@ -53,6 +53,11 @@ 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 requests
+        url = os.environ('GTRANSURL')
+        params = {'sl': sselected_language, 'tl': tselected_language, 'dt': 't', 'ie': 'UTF-8', 'oe': 'UTF-8', 'q': input_text}
+        translated_text = requests.get(url, params=params).text[0][0]
     elif model_name.startswith('t5'):
         prompt = f'translate {sselected_language} to {tselected_language}: {input_text}'
         print(prompt)