Update app.py

app.py

@@ -48,7 +48,7 @@ stopping_criteria = ListOfTokensStoppingCriteria(tokenizer, stop_tokens)
 # Añade tu criterio de parada a una StoppingCriteriaList
 stopping_criteria_list = StoppingCriteriaList([stopping_criteria])
 
-def generate_text(prompt, context, model, tokenizer, max_length=max_seq_length):
+def generate_text(prompt, context, max_length=max_seq_length):
   prompt=prompt.replace("\n", "").replace("¿","").replace("?","")
   input_text = f'''<bos><start_of_turn>system\n{context}<end_of_turn><start_of_turn>user\n{prompt}<end_of_turn><start_of_turn>model\n'''
   inputs = tokenizer.encode(input_text, return_tensors="pt", add_special_tokens=False).to("cuda:0")
@@ -62,9 +62,9 @@ def generate_text(prompt, context, model, tokenizer, max_length=max_seq_length):
   outputs = model.generate(generation_config=generation_config, input_ids=inputs, stopping_criteria=stopping_criteria_list,)
   return tokenizer.decode(outputs[0], skip_special_tokens=False) #True
 
-def mostrar_respuesta(pregunta, contexto, model, tokenizer):
+def mostrar_respuesta(pregunta, contexto):
     try:
-      res= generate_text(pregunta, contexto, model, tokenizer, max_length=max_seq_length)
+      res= generate_text(pregunta, contexto, max_length=max_seq_length)
       return str(res)
     except Exception as e:
       return str(e)