Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -65,18 +65,18 @@ def get_current_time_in_timezone(timezone: str) -> str:
|
|
65 |
final_answer = FinalAnswerTool()
|
66 |
|
67 |
# 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:
|
68 |
-
model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
|
69 |
-
|
70 |
-
model = HfApiModel(
|
71 |
-
max_tokens=2096,
|
72 |
-
temperature=0.5,
|
73 |
-
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
|
74 |
-
token = os.getenv("HF_TOKEN"),
|
75 |
-
custom_role_conversions=None,
|
76 |
-
)
|
77 |
|
78 |
|
79 |
-
|
80 |
|
81 |
# Import tool from Hub
|
82 |
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
|
|
|
65 |
final_answer = FinalAnswerTool()
|
66 |
|
67 |
# 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:
|
68 |
+
#model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
|
69 |
+
|
70 |
+
# model = HfApiModel(
|
71 |
+
# max_tokens=2096,
|
72 |
+
# temperature=0.5,
|
73 |
+
# model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
|
74 |
+
# token = os.getenv("HF_TOKEN"),
|
75 |
+
# custom_role_conversions=None,
|
76 |
+
# )
|
77 |
|
78 |
|
79 |
+
model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=os.getenv("GEMINI_API_KEY"))
|
80 |
|
81 |
# Import tool from Hub
|
82 |
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
|