jesoteric commited on
Commit
fc387c9
1 Parent(s): 379f47a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -11
app.py CHANGED
@@ -304,17 +304,8 @@ if __name__ == "__main__":
304
 
305
  outputs = []
306
 
307
- if inputs["task_type"] == 'det':
308
- outputs.append(gr.outputs.Image(type="pil", label="Output Image"))
309
- elif inputs["task_type"] in ['seg', 'automatic']:
310
- outputs.append(gr.outputs.Image(type="pil", label="Output Image"))
311
- outputs.append(gr.outputs.Image(type="pil", label="Output Mask"))
312
- elif inputs["task_type"] == 'inpainting':
313
- outputs.append(gr.outputs.Image(type="pil", label="Output Image"))
314
- outputs.append(gr.outputs.Image(type="pil", label="Output Mask"))
315
- else:
316
- print("task_type:{} error!".format(inputs["task_type"]))
317
-
318
 
319
  iface = gr.Interface(run_grounded_sam, inputs, outputs, title="Grounded SAM Demo")
320
 
 
304
 
305
  outputs = []
306
 
307
+ outputs.append(gr.outputs.Image(type="pil", label="Output Image"))
308
+ outputs.append(gr.outputs.Image(type="pil", label="Output Mask"))
 
 
 
 
 
 
 
 
 
309
 
310
  iface = gr.Interface(run_grounded_sam, inputs, outputs, title="Grounded SAM Demo")
311