Ailyth commited on
Commit
9b6cf76
·
verified ·
1 Parent(s): 37a7956

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ predicted_class=''
23
  #https://cloud.google.com/vertex-ai/docs/generative-ai/model-reference/gemini?hl=zh-cn
24
  MY_KEY= os.environ.get("GOOGLE_API_KEY")
25
  genai.configure(api_key=MY_KEY)
26
- gemini_model = genai.GenerativeModel('')
27
  neutral=os.environ.get("sys_info_0")
28
  toxic=os.environ.get("sys_info_1")
29
  heartfelt=os.environ.get("sys_info_2")
 
23
  #https://cloud.google.com/vertex-ai/docs/generative-ai/model-reference/gemini?hl=zh-cn
24
  MY_KEY= os.environ.get("GOOGLE_API_KEY")
25
  genai.configure(api_key=MY_KEY)
26
+ gemini_model = genai.GenerativeModel('gemini-pro-vision')
27
  neutral=os.environ.get("sys_info_0")
28
  toxic=os.environ.get("sys_info_1")
29
  heartfelt=os.environ.get("sys_info_2")