Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -39,11 +39,14 @@ final_answer = FinalAnswerTool()
|
|
| 39 |
# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
|
| 40 |
# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
|
| 41 |
|
|
|
|
|
|
|
|
|
|
| 42 |
|
| 43 |
model = LiteLLMModel(
|
| 44 |
model_id="gemini/gemini-2.0-flash-lite-001",
|
| 45 |
temperature=0.7,
|
| 46 |
-
api_key=
|
| 47 |
)
|
| 48 |
'''
|
| 49 |
model = HfApiModel(
|
|
|
|
| 39 |
# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
|
| 40 |
# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
|
| 41 |
|
| 42 |
+
import os
|
| 43 |
+
|
| 44 |
+
api_key = os.getenv("GEMINI_KEY")
|
| 45 |
|
| 46 |
model = LiteLLMModel(
|
| 47 |
model_id="gemini/gemini-2.0-flash-lite-001",
|
| 48 |
temperature=0.7,
|
| 49 |
+
api_key=api_key
|
| 50 |
)
|
| 51 |
'''
|
| 52 |
model = HfApiModel(
|