Update app.py

app.py

@@ -371,10 +371,11 @@ class Translators:
         tokenizer = AutoTokenizer.from_pretrained(self.model_name)
         model = AutoModelForSeq2SeqLM.from_pretrained(self.model_name)
         self.input_text = self.input_text if self.input_text.endswith('.') else f'{self.input_text}.'
-        inputs = tokenizer.encode(f"Translate to {self.tl}: {self.input_text}", return_tensors="pt")
+        inputs = tokenizer.encode(f"Translate to {self.tl}: {self.input_text} Translation:", return_tensors="pt")
         outputs = model.generate(inputs)
         translation = tokenizer.decode(outputs[0])
         translation = translation.replace('<pad> ', '').replace('</s>', '')
+        translation = translation.split('Translation:')[-1].strip() if 'Translation:' in translation else translation.strip()
         return translation
     
     def bloomz(self):