Update app.py

app.py

@@ -4,6 +4,8 @@ import gradio as gr
 os.system("git clone https://github.com/AK391/TokenCut.git")
 os.chdir("TokenCut")
 
+os.system("wget https://i.imgur.com/G4wpITa.jpg -O dog.jpg")
+
 def inference(img):
   os.system("python main_tokencut.py --image_path "+img+" --visualize all")
   return "./outputs/TokenCut-vit_small16_k/test_TokenCut_attn.jpg","./outputs/TokenCut-vit_small16_k/test_TokenCut_pred.jpg"
@@ -13,5 +15,5 @@ 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.jpeg']]
+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()