Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ from tools.final_answer import FinalAnswerTool
|
|
7 |
|
8 |
from Gradio_UI import GradioUI
|
9 |
|
10 |
-
|
11 |
|
12 |
@tool
|
13 |
def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
|
@@ -40,7 +40,8 @@ model = HfApiModel(
|
|
40 |
max_tokens=2096,
|
41 |
temperature=0.5,
|
42 |
model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',
|
43 |
-
custom_role_conversions=None
|
|
|
44 |
)
|
45 |
|
46 |
|
|
|
7 |
|
8 |
from Gradio_UI import GradioUI
|
9 |
|
10 |
+
TOKEN = os.getenv("HF_TOKEN")
|
11 |
|
12 |
@tool
|
13 |
def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
|
|
|
40 |
max_tokens=2096,
|
41 |
temperature=0.5,
|
42 |
model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',
|
43 |
+
custom_role_conversions=None,,
|
44 |
+
token=TOKEN
|
45 |
)
|
46 |
|
47 |
|