Update app.py

app.py

@@ -31,8 +31,8 @@ def translate(input, output, text):
     }
     response = requests.request("POST", API_URL,  json=json_) 
     output = response.json()[0]['generated_text']
-
-    return output.replace(instruction, '', 1).split('<end>')[0]
+    output = output.replace(instruction, '', 1)
+    return output[output.find(f'Translation in {output}')+len(f'Translation in {output}'):].split('<end>')[0]
 
 demo = gr.Blocks()