Update app.py

app.py

@@ -17,7 +17,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
 
 pipe = pipe.to(device)
 
-def text_to_image(text):
+def text_to_image(summary, image_name):
     # Crea una instancia del cliente de S3
     s3 = boto3.client('s3',
                   aws_access_key_id=AWS_ACCESS_KEY_ID,
@@ -35,10 +35,9 @@ def text_to_image(text):
         # Sube la imagen al bucket de S3
         s3.upload_fileobj(image_buffer, S3_BUCKET_NAME, s3_key)
     
-    def generator_image(text):
-        prompt = text
+    def generator_image(summary, image_name):
+        prompt = summary
         image = pipe(prompt).images[0]
-        image_name = '-'.join(prompt.split()) + ".webp"
         
         # Guarda la imagen en S3
         save_image_to_s3(image, image_name)
@@ -46,9 +45,8 @@ def text_to_image(text):
         return image_name
     
     # generate image
-    image_name = generator_image(text)
-    return f"{image_name}"
-    
+    image_name = '-'.join(image_name.split()) + ".webp"
+    return generator_image(summary, image_name)