JaphetHernandez commited on
Commit
85ec4d4
1 Parent(s): 091fb96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -1,5 +1,9 @@
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
2
 
 
 
 
 
3
  # Cargar el modelo y el tokenizer
4
  model_name = "nombre_del_modelo"
5
  model = AutoModelForCausalLM.from_pretrained(model_name)
 
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
2
 
3
+ # Token Secret of Hugging Face
4
+ huggingface_token = st.secrets["HUGGINGFACEHUB_API_TOKEN"]
5
+ login(huggingface_token)
6
+
7
  # Cargar el modelo y el tokenizer
8
  model_name = "nombre_del_modelo"
9
  model = AutoModelForCausalLM.from_pretrained(model_name)