Update app.py

app.py

@@ -7,7 +7,7 @@ from einops import rearrange
 import requests
 import spaces
 from huggingface_hub import login
-from gradio_imageslider import ImageSlider  # Import ImageSlider
+from gradio_imageslider import ImageSlider
 from diffusers.utils import load_image
 from diffusers import FluxControlNetPipeline, FluxControlNetModel
 
@@ -37,7 +37,18 @@ control_modes = {
     "lq": 6,
 }
 
+def load_and_convert_image(image):
+    if isinstance(image, str):
+        image = Image.open(image)
+    elif isinstance(image, bytes):
+        image = Image.open(io.BytesIO(image))
+    # Convert AVIF to PNG if necessary
+    if image.format == 'AVIF':
+        image = image.convert("RGB")  # Convert to a format PIL can handle
+    return image
+
 def preprocess_image(image, target_width, target_height, crop=True):
+    image = load_and_convert_image(image)
     if crop:
         original_width, original_height = image.size
 
@@ -103,4 +114,4 @@ interface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch(share=True)