Update app.py

app.py

@@ -41,13 +41,15 @@ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str
         f"{table_str}\n\n"
         "لطفاً بر اساس این اطلاعات دو پاراگراف مشاوره تخصصی بنویسید."
     )
-    # فراخوانی text_generation با پارامتر درست
-    outputs = client.text_generation(
-        model="google/gemma-2b",
-        prompt=prompt,
-        max_new_tokens=256
-    )
-    return outputs
+    try:
+        outputs = client.text_generation(
+            prompt=prompt,
+            model="google/gemma-2b-it",  # استفاده از مدل جدید
+            max_new_tokens=256
+        )
+        return outputs
+    except Exception as e:
+        return f"خطا در تولید مشاوره: {str(e)}"
 
 current_year = pd.Timestamp.now().year
 years = [str(y) for y in range(2000, current_year+1)]