Update app.py

app.py

@@ -20,31 +20,33 @@ model = AutoModelForCausalLM.from_pretrained(model_name)
 chat_history = []
 # تابع پاسخ‌دهی
 def student_bot(question):
-    chat_history.append(f"پرسش: {question}")
-    
-    prompt = """پرسش: زمان انتخاب واحد چه زمانی است؟
+    prompt = f"""پرسش: زمان انتخاب واحد چه زمانی است؟
 پاسخ: معمولا در پایان هر ترم تحصیلی، یعنی شهریور و بهمن انجام می‌شود.
 
 پرسش: زمان حذف و اضافه کی هست؟
 پاسخ: حدود یک هفته بعد از شروع ترم تحصیلی است.
-""" + "\n".join(chat_history[-3:]) + "\nپاسخ:"
-
-    for key in faq_dict:
-        if key in question:
-            return faq_dict[key]
 
+پرسش: {question}
+پاسخ:"""
 
     inputs = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(
         **inputs,
         max_new_tokens=60,
-        pad_token_id=tokenizer.eos_token_id
+        pad_token_id=tokenizer.eos_token_id,
+        repetition_penalty=1.8,
+        no_repeat_ngram_size=3,
+        temperature=0.7,
+        top_p=0.9,
+        top_k=50,
+        do_sample=True
     )
     answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return answer.split("پاسخ:")[-1].strip()
 
 
 
+
 # رابط Gradio
 gr.Interface(
     fn=student_bot,