Update app.py

app.py

@@ -34,15 +34,13 @@ def square_center_crop(img, target_size=768):
     return img_cropped.resize((target_size, target_size), Image.Resampling.LANCZOS)
 
 def duplicate_horizontally(img):
+    # Convert PIL Image to numpy array
     width, height = img.size
     if width != height:
         raise ValueError(f"Input image must be square, got {width}x{height}")
-        
-    # Convert to numpy array
+    
     img_array = np.array(img)
-    # Stack horizontally
-    duplicated = np.hstack((img_array, img_array))
-    # Convert back to PIL Image
+    duplicated = np.concatenate([img_array, img_array], axis=1)
     return Image.fromarray(duplicated)
 
 # Load the mask image