Update app.py

app.py

@@ -19,6 +19,10 @@ def generate_image(prompt):
         else:
             raise Exception("No images returned by the model.")
             
+    except IndexError as e:
+        print(f"Index error during image generation: {e}")
+        traceback.print_exc()
+        return None
     except Exception as e:
         print(f"Error generating image: {e}")
         traceback.print_exc()