rockdrigoma commited on
Commit
ed1bf18
·
verified ·
1 Parent(s): 7ef95f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -7,7 +7,7 @@ from tools.final_answer import FinalAnswerTool
7
 
8
  from Gradio_UI import GradioUI
9
 
10
- api_key = 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,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