Matteo-CNPPS commited on
Commit
4b1ac20
·
verified ·
1 Parent(s): e49d5ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -8
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import gradio as gr
2
 
3
  from smolagents import HfApiModel, Tool, CodeAgent
4
- from huggingface_hub import HfApi
5
 
6
  import os
7
  import sys
@@ -11,11 +10,6 @@ if './lib' not in sys.path :
11
  sys.path.append('./lib')
12
  from ingestion_chroma import retrieve_info_from_db
13
 
14
-
15
- hf_token = os.getenv("HF_TOKEN_all")
16
- api = HfApi()
17
- api.login(token=hf_token)
18
-
19
  ############################################################################################
20
  ################################### TOOLS ##################################################
21
  ############################################################################################
@@ -92,8 +86,8 @@ User's question : """
92
  yield agent_output
93
 
94
  ############################################################################################
95
-
96
- model = HfApiModel("Qwen/Qwen2.5-Coder-32B-Instruct")
97
 
98
  retriever_tool = Chroma_retrieverTool()
99
  get_ESRS_info_tool = ESRS_info_tool()
 
1
  import gradio as gr
2
 
3
  from smolagents import HfApiModel, Tool, CodeAgent
 
4
 
5
  import os
6
  import sys
 
10
  sys.path.append('./lib')
11
  from ingestion_chroma import retrieve_info_from_db
12
 
 
 
 
 
 
13
  ############################################################################################
14
  ################################### TOOLS ##################################################
15
  ############################################################################################
 
86
  yield agent_output
87
 
88
  ############################################################################################
89
+ hf_token = os.getenv("HF_TOKEN_all")
90
+ model = HfApiModel("Qwen/Qwen2.5-Coder-32B-Instruct", hf_token=hf_token)
91
 
92
  retriever_tool = Chroma_retrieverTool()
93
  get_ESRS_info_tool = ESRS_info_tool()