Update app.py

app.py

@@ -68,9 +68,12 @@ def create_image(image_pil,
             }
         #pipe.set_ip_adapter_scale(scale) ## Waiting for SD3 Diffuser integration
     
-    style_image = Image.open(image_pil).convert('RGB')
+    if hasattr(image_pil, 'read'):  # If it's a file
+        style_image = Image.open(image_pil).convert('RGB')
+    else:  # If it's already a PIL Image
+        style_image = image_pil.convert('RGB')
+    
     
-    print("starting inference")
     image = pipe(
             width=1024,
             height=1024,
@@ -83,12 +86,12 @@ def create_image(image_pil,
             ipadapter_scale=scale,
         ).images[0]
 
-    print("collecting")
+    
     if torch.cuda.is_available():
         torch.cuda.empty_cache()
         gc.collect()
 
-    print("printing image")
+    
     print(image)
     
     return image