Juliofc commited on
Commit
0bf6c87
verified
1 Parent(s): b0276a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -7,6 +7,11 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
7
  print(device)
8
  # Aseg煤rate de que tu token de Hugging Face est谩 cargado como una variable de entorno
9
  hf_token = os.environ.get("token")
 
 
 
 
 
10
 
11
  # Configuraci贸n inicial
12
  tokenizer = AutoTokenizer.from_pretrained("Juliofc/chaterapia_model")
 
7
  print(device)
8
  # Aseg煤rate de que tu token de Hugging Face est谩 cargado como una variable de entorno
9
  hf_token = os.environ.get("token")
10
+ if hf_token is not None:
11
+ from huggingface_hub import HfFolder
12
+ HfFolder.save_token(hf_token)
13
+ else:
14
+ print("No se encontr贸 el token de Hugging Face. Aseg煤rate de que la variable de entorno HF_TOKEN est茅 configurada.")
15
 
16
  # Configuraci贸n inicial
17
  tokenizer = AutoTokenizer.from_pretrained("Juliofc/chaterapia_model")