tistabaulopez commited on
Commit
c4255e2
verified
1 Parent(s): 8edb26f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -4,9 +4,11 @@ import threading
4
  import time
5
 
6
  # Cargar el modelo de lenguaje preentrenado
7
- model_name = "gpt2"
8
- tokenizer = AutoTokenizer.from_pretrained(model_name)
9
- model = AutoModelForCausalLM.from_pretrained(model_name)
 
 
10
 
11
  # Crear una funci贸n para comunicaci贸n en segundo plano
12
  def background_communication(response_log, stop_event):
 
4
  import time
5
 
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):