tistabaulopez commited on
Commit
01bded9
verified
1 Parent(s): c4255e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,9 +6,9 @@ import time
6
  # Cargar el modelo de lenguaje preentrenado
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
8
 
9
- model_name = "EleutherAI/gpt-neo-2.7B"
10
- tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
11
- model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=True)
12
 
13
  # Crear una funci贸n para comunicaci贸n en segundo plano
14
  def background_communication(response_log, stop_event):
 
6
  # Cargar el modelo de lenguaje preentrenado
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
8
 
9
+ model_name = "EleutherAI/gpt-neo-2.7B" # O cualquier otro modelo p煤blico como "gpt2"
10
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
11
+ model = AutoModelForCausalLM.from_pretrained(model_name)
12
 
13
  # Crear una funci贸n para comunicaci贸n en segundo plano
14
  def background_communication(response_log, stop_event):