Update app.py

app.py

@@ -8,14 +8,29 @@ model = AutoModelForCausalLM.from_pretrained(model_name)
 
 # تابع پاسخ‌دهی
 def student_bot(question):
-    inputs = tokenizer(question, return_tensors="pt")
+    prompt = f"""پرسش: زمان انتخاب واحد چه زمانی است؟
+پاسخ: معمولا در پایان هر ترم تحصیلی، یعنی شهریور و بهمن انجام می‌شود.
+
+پرسش: زمان حذف و اضافه کی هست؟
+پاسخ: حدود یک هفته بعد از شروع ترم تحصیلی است.
+
+پرسش: چه زمانی می‌توان انصراف از درس داد؟
+پاسخ: معمولاً تا هفته هشتم می‌توان درس را حذف کرد.
+
+پرسش: {question}
+پاسخ:"""
+
+    inputs = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(
         **inputs,
-        max_new_tokens=50,
+        max_new_tokens=60,
         pad_token_id=tokenizer.eos_token_id
     )
     answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return answer[len(question):].strip()
+    
+    # حذف prompt اولیه از متن خروجی
+    return answer.split("پاسخ:")[-1].strip()
+
 
 # رابط Gradio
 gr.Interface(