Update app.py

app.py

@@ -43,7 +43,10 @@ translated_textarea = st.text("")
 
 # Handle the submit button click
 if submit_button:
-    prompt = f'translate {sselected_language} to {tselected_language}: {input_text}'
+    if model_name.startswith('Helsinki-NLP'):
+        prompt = input_text
+    else:
+        prompt = f'translate {sselected_language} to {tselected_language}: {input_text}'
     print(prompt)
     input_ids = tokenizer.encode(prompt, return_tensors='pt')
     # Perform translation