vishnun commited on
Commit
e19e289
·
1 Parent(s): d6a546c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,9 +10,9 @@ dmodel = DetrForObjectDetection.from_pretrained('facebook/detr-resnet-50')
10
  model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
11
  processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
12
 
13
- i1 = gr.inputs.Image()
14
  i2 = gr.inputs.Textbox()
15
- o1 = gr.outputs.Image()
16
  o2 = gr.outputs.Textbox()
17
 
18
  def extract_image(image, text, num=1):
 
10
  model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
11
  processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
12
 
13
+ i1 = gr.inputs.Image(type="pil")
14
  i2 = gr.inputs.Textbox()
15
+ o1 = gr.outputs.Image(type="pil")
16
  o2 = gr.outputs.Textbox()
17
 
18
  def extract_image(image, text, num=1):