Update README.md

README.md

@@ -25,7 +25,7 @@ if tokenizer.pad_token is None:
     tokenizer.pad_token = tokenizer.eos_token
 
 # Define the inference function with TextStreamer
-def generate_answer_with_stream(model, tokenizer, text, max_new_tokens=350, temperature=0.5, top_k=40, top_p=0.9):
+def generate_answer_with_stream(model, tokenizer, text, max_new_tokens=1024, temperature=0.5, top_k=40, top_p=0.9):
     prompt = f"Answer the following question\n\n{text}\n\nQuestion:"
     
     # Tokenize the input text