Update app.py

app.py

@@ -66,18 +66,31 @@ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str
     print(prompt)
 
     try:
+        # آماده‌سازی ورودی برای مدل
         input_ids = tokenizer(prompt, return_tensors="pt").to("cuda")
+        # تولید خروجی
         outputs = model.generate(
             **input_ids,
             max_new_tokens=1024,
             do_sample=True,
-            temperature=0.7,
-            top_p=0.9
+            temperature=0.6,  # کاهش برای پاسخ‌های منسجم‌تر
+            top_p=0.85,      # تنظیم برای کیفیت بهتر
+            pad_token_id=tokenizer.eos_token_id  # جلوگیری از خطای pad token
         )
+        # دیکد کردن خروجی و حذف پرامپت
         generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        # حذف پرامپت از خروجی
+        if generated_text.startswith(prompt):
+            generated_text = generated_text[len(prompt):].strip()
+        
+        # بررسی اینکه خروجی خالی نباشد
+        if not generated_text:
+            return "مدل نتوانست پاسخ مناسبی تولید کند. لطفاً دوباره امتحان کنید."
+        
         print("خروجی مدل دریافت شد (به انگلیسی):")
         print(generated_text)
 
+        # ترجمه خروجی به فارسی
         translated_outputs = translator.translate(generated_text)
         print("خروجی ترجمه‌شده به فارسی:")
         print(translated_outputs)