Update app.py

app.py

@@ -41,7 +41,7 @@ def retrieve_answer(query, threshold=0.65, top_k=1):
 def generate_response_with_gpt2(answer, question):
     prompt = f"پاسخ به سوال: {question} بر اساس اطلاعات: {answer}"
     input_ids = tokenizer.encode(prompt, return_tensors="pt", truncation=True, max_length=512)
-    output_ids = gpt2_model.generate(input_ids, max_length=150, num_return_sequences=1, no_repeat_ngram_size=2, top_p=0.95, temperature=0.7)
+    output_ids = gpt2_model.generate(input_ids, max_new_tokens=150, num_return_sequences=1, no_repeat_ngram_size=2, top_p=0.95, temperature=0.7)
     return tokenizer.decode(output_ids[0], skip_special_tokens=True)
 
 # ===============================