Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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("
|
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, "
|
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' %}
|