Tzktz commited on
Commit
e1206d2
·
verified ·
1 Parent(s): 5f4da4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.open(original_image)
10
- mask_image = Image.open(mask_image).convert('L')
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)