Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -6,8 +6,8 @@ import gradio as gr
|
|
6 |
def inpaint_image(original_image, mask_image):
|
7 |
simple_lama = SimpleLama()
|
8 |
# Load images using PIL
|
9 |
-
original_image = Image.
|
10 |
-
mask_image = Image.
|
11 |
|
12 |
# Perform inpainting
|
13 |
result = simple_lama(original_image, mask_image)
|
|
|
6 |
def inpaint_image(original_image, mask_image):
|
7 |
simple_lama = SimpleLama()
|
8 |
# Load images using PIL
|
9 |
+
original_image = Image.fromarray(original_image)
|
10 |
+
mask_image = Image.fromarray(mask_image).convert('L')
|
11 |
|
12 |
# Perform inpainting
|
13 |
result = simple_lama(original_image, mask_image)
|