Liu Hong Yuan Tom commited on
Commit
294ae7d
·
verified ·
1 Parent(s): 22c7cd9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ print("google-generativeai:", genai.__version__)
12
  GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
13
 
14
  TITLE = """<h1 align="center">Gemini 1.5 Pro Playground 💬</h1>"""
15
- SUBTITLE = """<h2 align="center">Play with Gemini 1.5 Pro 0801 (the one that beat GPT-4o in Lmsys Leaderboard) and Gemini Pro Vision API</h2>"""
16
  DUPLICATE = """
17
  <div style="text-align: center; display: flex; justify-content: center; align-items: center;">
18
  <a href="https://huggingface.co/spaces/SkalskiP/ChatGemini?duplicate=true">
@@ -114,7 +114,7 @@ def bot(
114
  if chatbot[-1][0] and isinstance(chatbot[-1][0], str) \
115
  else []
116
  image_prompt = [Image.open(file).convert('RGB') for file in files]
117
- model = genai.GenerativeModel('gemini-pro-vision')
118
  response = model.generate_content(
119
  text_prompt + image_prompt,
120
  stream=True,
 
12
  GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
13
 
14
  TITLE = """<h1 align="center">Gemini 1.5 Pro Playground 💬</h1>"""
15
+ SUBTITLE = """<h2 align="center">Play with Gemini 1.5 Pro 0801 (the one that beat GPT-4o in Lmsys Leaderboard)</h2>"""
16
  DUPLICATE = """
17
  <div style="text-align: center; display: flex; justify-content: center; align-items: center;">
18
  <a href="https://huggingface.co/spaces/SkalskiP/ChatGemini?duplicate=true">
 
114
  if chatbot[-1][0] and isinstance(chatbot[-1][0], str) \
115
  else []
116
  image_prompt = [Image.open(file).convert('RGB') for file in files]
117
+ model = genai.GenerativeModel('gemini-1.5-pro-exp-0801')
118
  response = model.generate_content(
119
  text_prompt + image_prompt,
120
  stream=True,