Evaaaaa06 commited on
Commit
8e7c0cf
·
verified ·
1 Parent(s): 4307798

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -165,7 +165,8 @@ def handle_image_message(event):
165
  prompt = event.message.text
166
  store_user_message(user_id, "text", prompt)
167
  # 使用 Gemini 模型生成文字
168
- completion = model.start_chat(prompt, generation_config=generation_config)
 
169
  # 檢查生成結果是否為空
170
  if (completion.parts[0].text != None):
171
  # 取得生成結果
 
165
  prompt = event.message.text
166
  store_user_message(user_id, "text", prompt)
167
  # 使用 Gemini 模型生成文字
168
+ chat = model.start_chat(history = [])
169
+ completion = chat.send_message(prompt, generation_config=generation_config)
170
  # 檢查生成結果是否為空
171
  if (completion.parts[0].text != None):
172
  # 取得生成結果