Update README.md

README.md

@@ -63,5 +63,5 @@ for i in tokenizer.decode(generated_ids[0], skip_special_tokens=True).split('<en
 
 # Extract only the model's answer
 model_answer = ans.split("model")[1].strip()
-return model_answer
+print(model_answer)
 ```