Yangtao Wang
commited on
Commit
•
dc3edb7
1
Parent(s):
8aa2bcc
fix bug
Browse files
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="
|
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 |
|