uuu

app.py

@@ -63,12 +63,16 @@ pipe.init_ipadapter(
 # ----------------------------
 def preprocess_image(image_path):
     """Preprocess the input image for the pipeline."""
+    image = Image.open(image_path).convert("RGB")
+    # Ensure image is resized into a square based on the max dimension
+    size = max(image.size)
+    image = image.resize((size, size))
+
     preprocess = transforms.Compose([
         transforms.Resize((384, 384)),
         transforms.ToTensor(),
         transforms.ConvertImageDtype(torch.float16)
     ])
-    image = Image.open(image_path).convert('RGB')
     return preprocess(image).unsqueeze(0).to("cuda")
 
 # ----------------------------
@@ -130,4 +134,4 @@ interface = gr.Interface(
 # ----------------------------
 # Step 8: Launch Gradio App
 # ----------------------------
-interface.launch(share=True)
+interface.launch()