joackoEsp commited on
Commit
e3d3bd5
verified
1 Parent(s): 14cd230

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- inputs = tokenizer(f"Instrucci贸n: {system_prompt} \n Usuario: {prompt} \n Respuesta:", return_tensors="pt")
 
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,