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,
|