Hugging Face's logo

Spaces:
somosnlp
/
chaterapia_demo
Runtime error

App Files Community
Juliofc commited on
Commit
70aab85
·
verified ·
1 Parent(s): 540a670

Update app_1.py

Browse files
Files changed (1) hide show
  1. app_1.py +1 -1
app_1.py CHANGED
@@ -32,7 +32,7 @@ model_with_adapter = PeftModel.from_pretrained(model_base, "Juliofc/chaterapia_m
32
 
33
  def generate_text(input_text):
34
  # Codificar el texto de entrada
35
- input_ids = tokenizer.encode(input_text, return_tensors="pt")
36
 
37
  # Generar respuesta del modelo
38
  output_ids = model_with_adapter.generate(input_ids, max_length=50, num_return_sequences=1)
 
32
 
33
  def generate_text(input_text):
34
  # Codificar el texto de entrada
35
+ input_ids = tokenizer.encode(input_text, return_tensors="pt").to(device)
36
 
37
  # Generar respuesta del modelo
38
  output_ids = model_with_adapter.generate(input_ids, max_length=50, num_return_sequences=1)