Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,6 @@ import gradio as gr
|
|
2 |
from smolagents import CodeAgent, HfApiModel, Tool
|
3 |
from PIL import Image
|
4 |
import os
|
5 |
-
import spaces
|
6 |
|
7 |
# Define the image generation tool
|
8 |
image_generation_tool = Tool.from_space(
|
@@ -17,7 +16,6 @@ model = HfApiModel("Qwen/Qwen2.5-Coder-32B-Instruct") # Replace with your model
|
|
17 |
# Create the Smol Agent
|
18 |
agent = CodeAgent(tools=[image_generation_tool], model=model)
|
19 |
|
20 |
-
@spaces.GPU(duration=120)
|
21 |
# Define the chatbot function
|
22 |
def chat_with_agent(user_input, history):
|
23 |
if history is None:
|
|
|
2 |
from smolagents import CodeAgent, HfApiModel, Tool
|
3 |
from PIL import Image
|
4 |
import os
|
|
|
5 |
|
6 |
# Define the image generation tool
|
7 |
image_generation_tool = Tool.from_space(
|
|
|
16 |
# Create the Smol Agent
|
17 |
agent = CodeAgent(tools=[image_generation_tool], model=model)
|
18 |
|
|
|
19 |
# Define the chatbot function
|
20 |
def chat_with_agent(user_input, history):
|
21 |
if history is None:
|