Update app.py
Browse files
app.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
import json
|
| 3 |
import os
|
| 4 |
-
from smolagents import CodeAgent,
|
| 5 |
from PIL import Image
|
| 6 |
from transformers import pipeline
|
| 7 |
|
|
@@ -33,7 +33,7 @@ class VisionTool(Tool):
|
|
| 33 |
# Agent Initialization
|
| 34 |
|
| 35 |
hf_token = os.getenv("HF_TOKEN")
|
| 36 |
-
model =
|
| 37 |
|
| 38 |
agent = CodeAgent(
|
| 39 |
tools=[DuckDuckGoSearchTool(), VisitWebpageTool(), TranscriptionTool(), VisionTool()],
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
import json
|
| 3 |
import os
|
| 4 |
+
from smolagents import CodeAgent, HfModel, DuckDuckGoSearchTool, VisitWebpageTool, Tool
|
| 5 |
from PIL import Image
|
| 6 |
from transformers import pipeline
|
| 7 |
|
|
|
|
| 33 |
# Agent Initialization
|
| 34 |
|
| 35 |
hf_token = os.getenv("HF_TOKEN")
|
| 36 |
+
model = HfModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct",token=hf_token)
|
| 37 |
|
| 38 |
agent = CodeAgent(
|
| 39 |
tools=[DuckDuckGoSearchTool(), VisitWebpageTool(), TranscriptionTool(), VisionTool()],
|