Update app.py

app.py

@@ -15,7 +15,7 @@ def generate_image(prompt, prompt_name):
     try:
         print(f"Generating image for {prompt_name}")
         output = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0)
-        if output and hasattr(output, 'images') and len(output.images) > 0:
+        if output and hasattr(output, 'images') and output.images:
             print(f"Image generated for {prompt_name}")
             image = output.images[0]
             buffered = BytesIO()
@@ -23,7 +23,7 @@ def generate_image(prompt, prompt_name):
             image_bytes = buffered.getvalue()
             return image_bytes
         else:
-            print(f"No images found for {prompt_name}")
+            print(f"No images found or generated output is None for {prompt_name}")
             return None
     except Exception as e:
         print(f"An error occurred while generating image for {prompt_name}: {e}")