Update README.md

README.md

@@ -28,11 +28,11 @@ def predict(contesto, domanda):
 
     input_ids = tokenizer([prompt], return_tensors="pt").to(device)
 
-    generate_ids = model.generate(**input_ids, max_new_tokens=150, eos_token_id=0)
+    generate_ids = model.generate(**input_ids, max_new_tokens=150, eos_token_id=8112)
 
     answer = tokenizer.batch_decode(generate_ids)
     try:
-        final_answer = answer[0].split("##RISPOSTA: ")[1].split('\n', 1)[0]
+        final_answer = answer[0].split("##RISPOSTA: ")[1].split("##END")[0].strip("\n")
     except IndexError:
         final_answer = ""
     return final_answer