alanchen1115 commited on
Commit
9e380e6
·
verified ·
1 Parent(s): 5625061

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -4
app.py CHANGED
@@ -1,9 +1,9 @@
1
  import os
2
  import gradio as gr
3
- import google.generativeai as genai
4
 
5
  # 設定 Google AI API 金鑰
6
- genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
7
 
8
  # 連結雲端 Gemini flash model
9
  model = genai.GenerativeModel("gemini-2.0-flash")
@@ -12,10 +12,16 @@ model = genai.GenerativeModel("gemini-2.0-flash")
12
  def get_gemini_response(input, image):
13
  # 如果使用者輸入了文字,則使用文字和圖片一起生成回應
14
  if input != "":
15
- response = model.generate_content([input, image])
 
 
 
16
  # 否則只使用圖片生成回應
17
  else:
18
- response = model.generate_content(image)
 
 
 
19
  # 返回回應文字
20
  return response.text
21
 
 
1
  import os
2
  import gradio as gr
3
+ from google import genai
4
 
5
  # 設定 Google AI API 金鑰
6
+ client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))
7
 
8
  # 連結雲端 Gemini flash model
9
  model = genai.GenerativeModel("gemini-2.0-flash")
 
12
  def get_gemini_response(input, image):
13
  # 如果使用者輸入了文字,則使用文字和圖片一起生成回應
14
  if input != "":
15
+ response = client.models.generate_content(
16
+ model="gemini-2.0-flash",
17
+ contents=[image, input]
18
+ )
19
  # 否則只使用圖片生成回應
20
  else:
21
+ response = client.models.generate_content(
22
+ model="gemini-2.0-flash",
23
+ contents=[image, '說說這是什麼!']
24
+ )
25
  # 返回回應文字
26
  return response.text
27