Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,8 @@ model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float
|
|
11 |
def generate_response(prompt):
|
12 |
system_prompt = "Responde solo con el texto solicitado, sin informaci贸n personal ni datos irrelevantes."
|
13 |
|
14 |
-
|
|
|
15 |
|
16 |
outputs = model.generate(
|
17 |
inputs.input_ids,
|
|
|
11 |
def generate_response(prompt):
|
12 |
system_prompt = "Responde solo con el texto solicitado, sin informaci贸n personal ni datos irrelevantes."
|
13 |
|
14 |
+
inputs = tokenizer(f"Instrucci贸n: {system_prompt} \n Pregunta: {prompt} \n Respuesta directa:", return_tensors="pt")
|
15 |
+
|
16 |
|
17 |
outputs = model.generate(
|
18 |
inputs.input_ids,
|