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"]
+models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large", "Unbabel/TowerInstruct-7B-v0.2", "Unbabel/TowerInstruct-Mistral-7B-v0.2"]
 
 # Create two columns
 scol, tcol = st.columns(2)
@@ -62,7 +62,7 @@ if submit_button:
         # Decode the translated text
         translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
     else:
-        pipe = pipeline("text-generation", model="Unbabel/TowerInstruct-7B-v0.2", torch_dtype=torch.bfloat16, device_map="auto")
+        pipe = pipeline("text-generation", model=model_name, torch_dtype=torch.bfloat16, device_map="auto")
         # We use the tokenizer’s chat template to format each message - see https://huggingface.co/docs/transformers/main/en/chat_templating
         messages = [
             {"role": "user", "content": f"Translate the following text from {sselected_language} into {tselected_language}.\n{sselected_language}: {input_text}.\n{tselected_language}:"},