Update raw.py

raw.py

@@ -29,6 +29,7 @@ def generate_image(prompt, scale, steps, control_image):
     w, h = control_image.size
     # Upscale x1
     control_image = control_image.resize((int(w * scale), int(h * scale)))
+    print("Size to: " + control_image.size[0] + ", " + control_image.size[1])
     image = pipe(
         prompt=prompt,
         control_image=control_image,
@@ -50,7 +51,7 @@ iface = gr.Interface(
         gr.Image(type="pil", label="Control Image"),
     ],
     outputs=[
-        gr.Image(type="pil", label="Generated Image"),
+        gr.Image(type="pil", label="Generated Image", format="png"),
     ],
     title="FLUX ControlNet Image Generation",
     description="Generate images using the FluxControlNetPipeline. Upload a control image and enter a prompt to create an image.",