Update app.py

app.py

@@ -3,6 +3,7 @@ import torch
 import spaces
 from diffusers import FluxInpaintPipeline
 from PIL import Image, ImageFile
+import numpy as np
 
 #ImageFile.LOAD_TRUNCATED_IMAGES = True
 
@@ -36,11 +37,13 @@ def duplicate_horizontally(img):
     width, height = img.size
     if width != height:
         raise ValueError(f"Input image must be square, got {width}x{height}")
-
-    new_image = Image.new('RGB', (width * 2, height))
-    new_image.paste(img, (0, 0))
-    #new_image.paste(img, (width, 0))
-    return new_image
+        
+    # Convert to numpy array
+    img_array = np.array(img)
+    # Stack horizontally
+    duplicated = np.hstack((img_array, img_array))
+    # Convert back to PIL Image
+    return Image.fromarray(duplicated)
 
 # Load the mask image
 mask = Image.open("mask_square.png")
@@ -81,7 +84,7 @@ with gr.Blocks() as demo:
         with gr.Column():
             input_image = gr.Image(
                 label="Upload Logo Image",
-                type="pil",
+                type="numpy",
                 height=384
             )
             cropped_image = gr.Image(