Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -8,33 +8,13 @@ from smolagents import DuckDuckGoSearchTool, LiteLLMModel, PythonInterpreterTool
|
|
8 |
|
9 |
from litellm import Router
|
10 |
|
11 |
-
model_list = [
|
12 |
-
{
|
13 |
-
"model_name": "deepseek-r1-0528-key1",
|
14 |
-
"litellm_params": {
|
15 |
-
"model": "openrouter/deepseek/deepseek-r1-0528:free",
|
16 |
-
"api_key": "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",
|
17 |
-
"max_retries": 3
|
18 |
-
}
|
19 |
-
},
|
20 |
-
{
|
21 |
-
"model_name": "deepseek-r1-0528-key2",
|
22 |
-
"litellm_params": {
|
23 |
-
"model": "openrouter/deepseek/deepseek-r1-0528:free",
|
24 |
-
"api_key": "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",
|
25 |
-
"max_retries": 3
|
26 |
-
}
|
27 |
-
},
|
28 |
-
# Add more keys/deployments as needed
|
29 |
-
]
|
30 |
-
|
31 |
|
32 |
|
33 |
#model = HfApiModel(
|
34 |
# model_id="Qwen/Qwen3-235B-A22B",
|
35 |
# api_key=os.getenv("HFT") # Ensure this environment variable is set
|
36 |
#)
|
37 |
-
model = LiteLLMModel(model_id="together_ai/Qwen/Qwen3-235B-A22B-fp8-tput", api_key="
|
38 |
|
39 |
|
40 |
#model = Router(model_list=model_list)
|
|
|
8 |
|
9 |
from litellm import Router
|
10 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
|
12 |
|
13 |
#model = HfApiModel(
|
14 |
# model_id="Qwen/Qwen3-235B-A22B",
|
15 |
# api_key=os.getenv("HFT") # Ensure this environment variable is set
|
16 |
#)
|
17 |
+
model = LiteLLMModel(model_id="together_ai/Qwen/Qwen3-235B-A22B-fp8-tput", api_key=os.getenv("HFT"),max_retries=3)
|
18 |
|
19 |
|
20 |
#model = Router(model_list=model_list)
|