Update app.py

app.py

@@ -266,6 +266,8 @@ class Translators:
         inputs = tokenizer(prompt, return_tensors="pt")
         outputs = model.generate(**inputs, max_new_tokens=512)
         output = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        print(output)
+        # result = output.rsplit(f'{self.tl}:')[-1].strip() if f'{self.tl}:' in output else output.strip()
         result = output.rsplit(f'{self.tl}:')[-1].strip() if '\n' in output or f'{self.tl}:' in output else output.strip()
         return result