Spaces:
Sleeping
Sleeping
Vlad Bastina
commited on
Commit
·
e601c67
1
Parent(s):
3d01d70
swap model for flash
Browse files- query_chat.py +1 -1
query_chat.py
CHANGED
|
@@ -47,7 +47,7 @@ First Brochure:
|
|
| 47 |
Second Brochure:
|
| 48 |
{self.text2}
|
| 49 |
'''
|
| 50 |
-
return genai.GenerativeModel("gemini-
|
| 51 |
|
| 52 |
def answer_question(self, question: str) -> str:
|
| 53 |
"""Generates a response by including conversation history in the prompt."""
|
|
|
|
| 47 |
Second Brochure:
|
| 48 |
{self.text2}
|
| 49 |
'''
|
| 50 |
+
return genai.GenerativeModel("gemini-2.0-flash", system_instruction=system_instruction)
|
| 51 |
|
| 52 |
def answer_question(self, question: str) -> str:
|
| 53 |
"""Generates a response by including conversation history in the prompt."""
|