Update app.py

app.py

@@ -19,4 +19,4 @@ description="Gradio demo for TokenCut: Self-Supervised Transformers for Unsuperv
 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>"
 
 examples=[['dog.jpg']]
-gr.Interface(inference,gr.inputs.Image(type="filepath"),[gr.outputs.Image(type="file"),gr.outputs.Image(type="file")],title=title,description=description,article=article,examples=examples).launch()
+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)