not-lain commited on
Commit
43b320b
·
verified ·
1 Parent(s): 966063a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -37,10 +37,12 @@ def fn(image):
37
  pred_pil = transforms.ToPILImage()(pred)
38
  mask = pred_pil.resize(image_size)
39
  image.putalpha(mask)
40
- return (image, origin)
 
41
 
42
 
43
- slider1 = ImageSlider(label="birefnet", type="pil")
 
44
  slider2 = ImageSlider(label="birefnet", type="pil")
45
  image = gr.Image(label="Upload an image")
46
  text = gr.Textbox(label="Paste an image URL")
@@ -50,7 +52,7 @@ chameleon = load_img("chameleon.jpg", output_type="pil")
50
 
51
  url = "https://hips.hearstapps.com/hmg-prod/images/gettyimages-1229892983-square.jpg"
52
  tab1 = gr.Interface(
53
- fn, inputs=image, outputs=slider1, examples=[chameleon], api_name="image"
54
  )
55
 
56
  tab2 = gr.Interface(fn, inputs=text, outputs=slider2, examples=[url], api_name="text")
 
37
  pred_pil = transforms.ToPILImage()(pred)
38
  mask = pred_pil.resize(image_size)
39
  image.putalpha(mask)
40
+ # return (image, origin)
41
+ return image
42
 
43
 
44
+ # slider1 = ImageSlider(label="birefnet", type="pil")
45
+ img1 = gr.Image(type= "pil", image_mode="RGBA")
46
  slider2 = ImageSlider(label="birefnet", type="pil")
47
  image = gr.Image(label="Upload an image")
48
  text = gr.Textbox(label="Paste an image URL")
 
52
 
53
  url = "https://hips.hearstapps.com/hmg-prod/images/gettyimages-1229892983-square.jpg"
54
  tab1 = gr.Interface(
55
+ fn, inputs=image, outputs=img1, examples=[chameleon], api_name="image"
56
  )
57
 
58
  tab2 = gr.Interface(fn, inputs=text, outputs=slider2, examples=[url], api_name="text")