Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import requests
|
|
4 |
import inspect
|
5 |
import pandas as pd
|
6 |
#--SM
|
7 |
-
from smolagents import GoogleSearchTool,OpenAIServerModel,CodeAgent,DuckDuckGoSearchTool,VisitWebpageTool,HfApiModel
|
8 |
from smolagents.utils import encode_image_base64, make_image_url
|
9 |
import datetime
|
10 |
import requests
|
@@ -19,11 +19,10 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
|
19 |
#model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/"
|
20 |
#model_id="Qwen/Qwen2.5-Coder-32B-Instruct"
|
21 |
model_id="deepseek-ai/DeepSeek-Coder-V2-Lite-Base"
|
22 |
-
|
23 |
-
|
24 |
|
25 |
# Initialize the model
|
26 |
-
model = HfApiModel(model_id)
|
27 |
#model = HfApiModel("Qwen/Qwen2.5-Coder-32B-Instruct")
|
28 |
#model = HfApiModel("deepseek-ai/DeepSeek-R1", provider="together", max_tokens=8096)
|
29 |
|
|
|
4 |
import inspect
|
5 |
import pandas as pd
|
6 |
#--SM
|
7 |
+
from smolagents import GoogleSearchTool,OpenAIServerModel,CodeAgent,DuckDuckGoSearchTool,VisitWebpageTool,HfApiModel,LiteLLMModel
|
8 |
from smolagents.utils import encode_image_base64, make_image_url
|
9 |
import datetime
|
10 |
import requests
|
|
|
19 |
#model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/"
|
20 |
#model_id="Qwen/Qwen2.5-Coder-32B-Instruct"
|
21 |
model_id="deepseek-ai/DeepSeek-Coder-V2-Lite-Base"
|
22 |
+
model_=LiteLLMModel(model_id)
|
|
|
23 |
|
24 |
# Initialize the model
|
25 |
+
#model = HfApiModel(model_id)
|
26 |
#model = HfApiModel("Qwen/Qwen2.5-Coder-32B-Instruct")
|
27 |
#model = HfApiModel("deepseek-ai/DeepSeek-R1", provider="together", max_tokens=8096)
|
28 |
|