Update app.py

app.py

@@ -34,9 +34,9 @@ def eurollm(model_id, sl, tl, input_text):
     model = AutoModelForCausalLM.from_pretrained(model_id)  
     prompt = f"{sl}: {input_text}. {tl}:"
     inputs = tokenizer(prompt, return_tensors="pt")
-    outputs = model.generate(**inputs, max_new_tokens=20)
+    outputs = model.generate(**inputs, max_new_tokens=512)
     output = tokenizer.decode(outputs[0], skip_special_tokens=True) 
-    print(output)
+    print(output.rsplit(f'{tl}:')[-1])
     return output
 
 @spaces.GPU