Split MiLMMT result

app.py

@@ -145,7 +145,7 @@ class Translators:
         text = f"Translate this from {self.sl} to {self.tl}:\n{self.sl}: {self.input_text}\n{self.tl}:"
         inputs = tokenizer(text, add_special_tokens=False, return_tensors="pt")
         outputs = model.generate(**inputs, max_new_tokens=self.max_new_tokens)
-        return tokenizer.decode(outputs[0], skip_special_tokens=True)
+        return tokenizer.decode(outputs[0], skip_special_tokens=True).rsplit(f'{self.tl}:')[-1].strip()
     
     def translategemma(self):
         from huggingface_hub import login