Tom
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,8 +11,8 @@ print("google-generativeai:", genai.__version__)
|
|
| 11 |
|
| 12 |
GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
|
| 13 |
|
| 14 |
-
TITLE = """<h1 align="center">Gemini Playground 💬</h1>"""
|
| 15 |
-
SUBTITLE = """<h2 align="center">Play with Gemini Pro 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">
|
|
@@ -121,7 +121,7 @@ def bot(
|
|
| 121 |
generation_config=generation_config)
|
| 122 |
else:
|
| 123 |
messages = preprocess_chat_history(chatbot)
|
| 124 |
-
model = genai.GenerativeModel('gemini-pro')
|
| 125 |
response = model.generate_content(
|
| 126 |
messages,
|
| 127 |
stream=True,
|
|
|
|
| 11 |
|
| 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 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">
|
|
|
|
| 121 |
generation_config=generation_config)
|
| 122 |
else:
|
| 123 |
messages = preprocess_chat_history(chatbot)
|
| 124 |
+
model = genai.GenerativeModel('gemini-1.5-pro-latest')
|
| 125 |
response = model.generate_content(
|
| 126 |
messages,
|
| 127 |
stream=True,
|