Update app.py

app.py

@@ -17,23 +17,25 @@ def translate(text, source_lang, target_lang):
     
     instruction = f"Translate the text from {source_lang} to {target_lang}"
     
-    input_text = f"""### Instruction: {instruction}
+    input_text = f"""
+    ### Instruction: {instruction}
 
-### Input: {text}
+    ### Input: {text}
 
-### Response:
+    ### Response:
 """
     
     output = query({
         "inputs": input_text,
         "parameters": {
+            "return_text": False,
             "return_full_text": False
         }
     })
     
     # Extract the translated text from the model output
     # Assuming the model returns just the translation without the instruction format
-    translated_text = output[0]['generated_text'].strip()
+    translated_text = output[0]['generated_text']
     
     return translated_text