M4sterStudy commited on
Commit
7971d45
·
verified ·
1 Parent(s): f840d30

cofigurar token

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -1,8 +1,14 @@
1
- import gradio as gr
 
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
 
 
 
 
 
3
 
4
  # Cargar el modelo y el tokenizador
5
- model_name = "meta-llama/Meta-Llama-3-8B" # Este es el nombre del repositorio en Hugging Face
6
  tokenizer = AutoTokenizer.from_pretrained(model_name)
7
  model = AutoModelForCausalLM.from_pretrained(model_name)
8
 
 
1
+ import os
2
+ from huggingface_hub import login
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
+ import gradio as gr
5
+
6
+ # Autenticar usando el token almacenado como secreto
7
+ hf_token = os.getenv("HF_API_TOKEN")
8
+ login(hf_token)
9
 
10
  # Cargar el modelo y el tokenizador
11
+ model_name = "meta-llama/Meta-Llama-3-8B"
12
  tokenizer = AutoTokenizer.from_pretrained(model_name)
13
  model = AutoModelForCausalLM.from_pretrained(model_name)
14