eugpal4 commited on
Commit
17f9bea
·
verified ·
1 Parent(s): 2f60183

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +6 -4
main.py CHANGED
@@ -5,12 +5,14 @@ import requests
5
  from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, InferenceClientModel
6
  from tools import duckduckgo_search, wikipedia_search, summarize_text, load_memory, save_memory
7
 
8
- print(os.environ.get('HUGGINGFACEHUB_API_TOKEN'))
9
 
 
 
 
10
  # Token di Autorizzazione (verifica ambiente)
11
- #token = os.getenv('HF_HOME/token')
12
- #if not token:
13
- # raise ValueError("Imposta la variabile d'ambiente HF_TOKEN")
14
 
15
  # Scegli il modello
16
  model = HfApiModel(model_id="google/flan-t5-base", token=token)
 
5
  from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, InferenceClientModel
6
  from tools import duckduckgo_search, wikipedia_search, summarize_text, load_memory, save_memory
7
 
 
8
 
9
+ print(os.environ.get('HUGGINGFACEHUB_API_TOKEN')
10
+
11
+ # token = s.getenv('HF_HOME/token')
12
  # Token di Autorizzazione (verifica ambiente)
13
+ token = 'HUGGINGFACEHUB_API_TOKEN'
14
+ if not token:
15
+ raise ValueError("Imposta la variabile d'ambiente HF_TOKEN")
16
 
17
  # Scegli il modello
18
  model = HfApiModel(model_id="google/flan-t5-base", token=token)