Update gen.py

gen.py

@@ -148,8 +148,7 @@ prompt = (
 
 def generate(event):
     # Generar el texto usando el modelo
-    outputs = pipe(prompt+" "+event, max_new_tokens=1024, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
+    outputs = pipe(prompt+" "+event, max_new_tokens=512, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
 
     # Imprimir la salida generada
-    print(outputs[0]["generated_text"])
-    return outputs[0]["generated_text"]
+    return outputs