Spaces:
Runtime error
Runtime error
Commit
•
4c7eeb8
1
Parent(s):
e7ee165
Create app.py
Browse files
app.py
ADDED
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import os
|
2 |
+
import gradio as gr
|
3 |
+
from pathlib import Path
|
4 |
+
|
5 |
+
|
6 |
+
os.system("git clone https://github.com/YangtaoWANG95/TokenCut.git")
|
7 |
+
os.chdir("TokenCut")
|
8 |
+
|
9 |
+
os.system("wget https://raw.githubusercontent.com/YangtaoWANG95/TokenCut/master/examples/VOC07_000064.jpg -O parrot.jpg")
|
10 |
+
|
11 |
+
def inference(img):
|
12 |
+
os.system("python main_tokencut.py --image_path "+img+" --visualize all --resize 480")
|
13 |
+
filename = Path(img).stem
|
14 |
+
return "./outputs/TokenCut-vit_small16_k/"+filename+"_TokenCut_attn.jpg","./outputs/TokenCut-vit_small16_k/"+filename+"_TokenCut_pred.jpg"
|
15 |
+
|
16 |
+
title="TokenCut"
|
17 |
+
description="Gradio demo for TokenCut: Self-Supervised Transformers for Unsupervised Object Discovery using Normalized Cut. To use it, simply upload your image or click on one of the examples to load them. We resize the smaller edge of the image to 480 accelerate inference time. Read more at the links below"
|
18 |
+
|
19 |
+
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>"
|
20 |
+
|
21 |
+
examples=[['parrot.jpg']]
|
22 |
+
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)
|