Update app.py

app.py

@@ -18,20 +18,27 @@ pipe.load_lora_weights(
     weight_name="visual-identity-design.safetensors"
 )
 
-def square_center_crop(img, target_size=768):
+def square_center_crop_numpy(img, target_size=768):
     if img.mode in ('RGBA', 'P'):
         img = img.convert('RGB')
-
-    width, height = img.size
+    
+    # Convert PIL image to numpy array
+    img_array = np.array(img)
+    
+    # Get dimensions
+    height, width = img_array.shape[:2]
     crop_size = min(width, height)
-
+    
+    # Calculate crop coordinates
     left = (width - crop_size) // 2
     top = (height - crop_size) // 2
-    right = left + crop_size
-    bottom = top + crop_size
-
-    img_cropped = img.crop((left, top, right, bottom))
-    return img_cropped.resize((target_size, target_size), Image.Resampling.LANCZOS)
+    
+    # Perform the crop on numpy array
+    img_cropped = img_array[top:top+crop_size, left:left+crop_size]
+    
+    # Convert back to PIL and resize
+    img_pil = Image.fromarray(img_cropped)
+    return img_pil.resize((target_size, target_size), Image.Resampling.LANCZOS)
 
 def duplicate_horizontally(img):
     # Convert PIL Image to numpy array