diginoron commited on
Commit
0e8609b
·
verified ·
1 Parent(s): 8ba43cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -42,12 +42,13 @@ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str
42
  "لطفاً بر اساس این اطلاعات دو پاراگراف مشاوره تخصصی بنویسید."
43
  )
44
  try:
45
- outputs = client.text_generation(
46
- prompt=prompt,
47
- model="google/gemma-2b-it", # استفاده از مدل جدید
 
48
  max_new_tokens=256
49
  )
50
- return outputs
51
  except Exception as e:
52
  return f"خطا در تولید مشاوره: {str(e)}"
53
 
 
42
  "لطفاً بر اساس این اطلاعات دو پاراگراف مشاوره تخصصی بنویسید."
43
  )
44
  try:
45
+ # استفاده از متد conversational
46
+ outputs = client.conversational(
47
+ text=prompt,
48
+ model="google/gemma-2b-it",
49
  max_new_tokens=256
50
  )
51
+ return outputs["generated_text"]
52
  except Exception as e:
53
  return f"خطا در تولید مشاوره: {str(e)}"
54