Update gen.py

gen.py

@@ -153,9 +153,9 @@ prompt = (
 
 def generate(event):
     # Generar el texto usando el modelo
-    prompt = [{'role':'system','content':event},{'role': 'user', 'content': 'Which famous math number begins with 1.6 ...?'}]
+    prompt_msg = [{'role':'system','content':prompt},{'role': 'user', 'content': event}]
     inputs = tokenizer.apply_chat_template(
-        prompt,
+        prompt_msg,
         add_generation_prompt=True,
         return_tensors='pt'
     )