Juliofc commited on
Commit
e3b610a
verified
1 Parent(s): 91dc4cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,10 +24,10 @@ else:
24
  print("No se encontr贸 el token de Hugging Face. Aseg煤rate de que la variable de entorno HF_TOKEN est茅 configurada.")
25
 
26
  # Configuraci贸n inicial
27
- tokenizer = AutoTokenizer.from_pretrained("Juliofc/chaterapia_model")
28
  model_base = AutoModelForCausalLM.from_pretrained("google/gemma-2b-it").to(device)
29
  model_base.resize_token_embeddings(len(tokenizer))
30
- model_with_adapter = PeftModel.from_pretrained(model_base, "Juliofc/chaterapia_model").to(device)
31
 
32
  CHAT_TEMPLATE= """{% for message in messages %}
33
  {% if message['role'] == 'user' %}
 
24
  print("No se encontr贸 el token de Hugging Face. Aseg煤rate de que la variable de entorno HF_TOKEN est茅 configurada.")
25
 
26
  # Configuraci贸n inicial
27
+ tokenizer = AutoTokenizer.from_pretrained("somosnlp/chaterapia_model")
28
  model_base = AutoModelForCausalLM.from_pretrained("google/gemma-2b-it").to(device)
29
  model_base.resize_token_embeddings(len(tokenizer))
30
+ model_with_adapter = PeftModel.from_pretrained(model_base, "somosnlp/chaterapia_model").to(device)
31
 
32
  CHAT_TEMPLATE= """{% for message in messages %}
33
  {% if message['role'] == 'user' %}