Yangtao Wang commited on
Commit
dc3edb7
1 Parent(s): 8aa2bcc
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,5 +18,5 @@ description="Gradio demo for TokenCut: Self-Supervised Transformers for Unsuperv
18
  article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2202.11539' target='_blank'>Self-Supervised Transformers for Unsupervised Object Discovery using Normalized Cut</a> | <a href='https://github.com/YangtaoWANG95/TokenCut' target='_blank'>Github Repo</a></p>"
19
 
20
  examples=[['parrot.jpg']]
21
- gr.Interface(inference,gr.inputs.Image(type="filepath"),[gr.outputs.Image(type="file",label="TokenCut_attn"),gr.outputs.Image(type="file",label="TokenCut_predication")],title=title,description=description,article=article,examples=examples).launch(enable_queue=True)
22
 
 
18
  article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2202.11539' target='_blank'>Self-Supervised Transformers for Unsupervised Object Discovery using Normalized Cut</a> | <a href='https://github.com/YangtaoWANG95/TokenCut' target='_blank'>Github Repo</a></p>"
19
 
20
  examples=[['parrot.jpg']]
21
+ gr.Interface(inference,gr.inputs.Image(type="filepath"),[gr.outputs.Image(type="filepath",label="TokenCut_attn"),gr.outputs.Image(type="filepath",label="TokenCut_predication")],title=title,description=description,article=article,examples=examples).launch(enable_queue=True)
22