Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,10 +8,11 @@ import re
|
|
| 8 |
import os
|
| 9 |
|
| 10 |
hf_token = os.getenv("HF_TOKEN")
|
|
|
|
| 11 |
|
| 12 |
@st.cache_resource
|
| 13 |
def get_inference_client():
|
| 14 |
-
return InferenceClient("
|
| 15 |
|
| 16 |
def extract_and_clean_json(text):
|
| 17 |
json_match = re.search(r'\{.*\}', text, re.DOTALL)
|
|
|
|
| 8 |
import os
|
| 9 |
|
| 10 |
hf_token = os.getenv("HF_TOKEN")
|
| 11 |
+
gemini_token=os.getenv("GEMINI_TOKEN")
|
| 12 |
|
| 13 |
@st.cache_resource
|
| 14 |
def get_inference_client():
|
| 15 |
+
return InferenceClient("gemini-2.0-flash")
|
| 16 |
|
| 17 |
def extract_and_clean_json(text):
|
| 18 |
json_match = re.search(r'\{.*\}', text, re.DOTALL)
|