Update app.py

app.py

@@ -20,4 +20,4 @@ def query(payload):
 def generate_image(prompt):
     image_bytes = query({"inputs": prompt})
     image = Image.open(io.BytesIO(image_bytes))
-    return image
+    return image