Update app.py

app.py

@@ -48,12 +48,14 @@ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str
     print("پرامپت ساخته‌شده:")
     print(prompt)
     try:
-        print("در حال فراخوانی مدل meta-llama/Llama-2-7b-chat...")
-        outputs = client.text_generation(
-            prompt=prompt,
-            model="meta-llama/Llama-2-7b-chat",  # مدل جدید
+        print("در حال فراخوانی مدل google/gemma-2b-it با متد conversational...")
+        # استفاده از متد conversational
+        conversation = client.conversational(
+            text=prompt,
+            model="google/gemma-2b-it",
             max_new_tokens=1024
         )
+        outputs = conversation['generated_text'] if 'generated_text' in conversation else conversation
         print("خروجی مدل دریافت شد (به انگلیسی):")
         print(outputs)