Update app.py

app.py

@@ -19,7 +19,7 @@ def generate_image(prompt):
     try:
         with torch.no_grad():
             image = image_model(prompt).images[0]
-        return image
+        return image, None
     except Exception as e:
         return None, f"Error generating image: {str(e)}"