fix: build

app.py

@@ -1,16 +1,30 @@
 import gradio as gr
-import spaces
+
+# import spaces
 import torch
 from PIL import Image
+import numpy as np
+import cv2
 from simple_lama_inpainting import SimpleLama
 
-zero = torch.Tensor([0]).cuda()
-print(zero.device) # <-- 'cpu' 🤔
+zero = torch.Tensor([0])
+print(zero.device)  # <-- 'cpu' 🤔
 lama: SimpleLama = SimpleLama(device=zero.device)
 
-@spaces.GPU
-def lama_inpainting(image: Image, mask: Image) -> Image:
-    return lama(image, mask)
 
-inpaint = gr.Interface(fn=lama_inpainting, inputs=[gr.Image(), gr.ImageMask()], outputs=gr.Image())
+# @spaces.GPU
+def lama_inpainting(image: np.ndarray, mask: np.ndarray) -> np.ndarray:
+    print(image.shape, mask.shape)
+    res = lama(
+        Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB)).convert("RGB"),
+        Image.fromarray(cv2.cvtColor(mask, cv2.COLOR_BGR2RGB)).convert("L"),
+    )
+    return cv2.cvtColor(np.array(res), cv2.COLOR_RGB2BGR)
+
+
+inpaint = gr.Interface(
+    fn=lama_inpainting,
+    inputs=[gr.Image(label="image"), gr.Image(label="mask")],
+    outputs=gr.Image(),
+)
 inpaint.launch()