Juliofc commited on
Commit
ebb67d6
verified
1 Parent(s): 6b8bf19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -1,6 +1,16 @@
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
2
  from peft import PeftModel
3
  import gradio as gr
 
 
 
 
 
 
 
 
 
 
4
 
5
 
6
  # Cargar el tokenizador
 
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
2
  from peft import PeftModel
3
  import gradio as gr
4
+ import os
5
+
6
+ # Aseg煤rate de que tu token de Hugging Face est谩 cargado como una variable de entorno
7
+ hf_token = os.environ.get("token")
8
+
9
+ if hf_token is not None:
10
+ from huggingface_hub import HfFolder
11
+ HfFolder.save_token(hf_token)
12
+ else:
13
+ print("No se encontr贸 el token de Hugging Face. Aseg煤rate de que la variable de entorno HF_TOKEN est茅 configurada.")
14
 
15
 
16
  # Cargar el tokenizador