Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,8 @@ def predict(prompt):
|
|
30 |
}
|
31 |
|
32 |
model = genai.GenerativeModel(
|
33 |
-
model_name="gemini-
|
|
|
34 |
generation_config=generation_config,
|
35 |
)
|
36 |
|
@@ -39,8 +40,8 @@ def predict(prompt):
|
|
39 |
]
|
40 |
)
|
41 |
|
42 |
-
response = chat_session.send_message(prompt
|
43 |
-
|
44 |
return response.text
|
45 |
|
46 |
# Create the Gradio interface
|
|
|
30 |
}
|
31 |
|
32 |
model = genai.GenerativeModel(
|
33 |
+
model_name="gemini-1.5-pro",
|
34 |
+
#model_name="gemini-2.0-flash-exp",
|
35 |
generation_config=generation_config,
|
36 |
)
|
37 |
|
|
|
40 |
]
|
41 |
)
|
42 |
|
43 |
+
#response = chat_session.send_message(prompt)
|
44 |
+
response = model.generate_content(contents=prompt, tools='google_search_retrieval')
|
45 |
return response.text
|
46 |
|
47 |
# Create the Gradio interface
|