alanchen1115 commited on
Commit
abe0b24
·
verified ·
1 Parent(s): 11276ed

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -82,7 +82,7 @@ def handle_message(event):
82
  # if not response.parts or not response.parts[0].text:
83
  # out = response.parts[0].text
84
  message = event.message.text
85
- out = model.generate_content(message)
86
 
87
  except:
88
  out = "Gemini error!"
 
82
  # if not response.parts or not response.parts[0].text:
83
  # out = response.parts[0].text
84
  message = event.message.text
85
+ out = client.predict(message, api_name="/predict")
86
 
87
  except:
88
  out = "Gemini error!"