Spaces:
Sleeping
Sleeping
Update gemini_call.py
Browse files- gemini_call.py +1 -1
gemini_call.py
CHANGED
|
@@ -66,7 +66,7 @@ final_prompt = f'''Task:
|
|
| 66 |
|
| 67 |
api_key = os.getenv("GOOGLE_API_KEY")
|
| 68 |
genai.configure(api_key=api_key)
|
| 69 |
-
model = genai.GenerativeModel("gemini-2.0-flash
|
| 70 |
|
| 71 |
def extract_histogram_data(text):
|
| 72 |
"""Searches for reges to extract the info required for building the histogram"""
|
|
|
|
| 66 |
|
| 67 |
api_key = os.getenv("GOOGLE_API_KEY")
|
| 68 |
genai.configure(api_key=api_key)
|
| 69 |
+
model = genai.GenerativeModel("gemini-2.0-flash" , system_instruction=final_prompt)
|
| 70 |
|
| 71 |
def extract_histogram_data(text):
|
| 72 |
"""Searches for reges to extract the info required for building the histogram"""
|