Update gradio_app.py

gradio_app.py

@@ -23,6 +23,7 @@ from demo.visualizer import Visualizer, ColorMode
 
 import gradio as gr
 from huggingface_hub import hf_hub_download
+
 KEY_DICT = {"Cityscapes (19 classes)": "cityscapes",}
 
 SWIN_CFG_DICT = {"cityscapes": "configs/cityscapes/oneformer_swin_large_IN21k_384_bs16_90k.yaml"}
@@ -30,6 +31,8 @@ SWIN_CFG_DICT = {"cityscapes": "configs/cityscapes/oneformer_swin_large_IN21k_38
 SWIN_MODEL_DICT = {"cityscapes": hf_hub_download(repo_id="rubikk179/oneformer_cityscapes_Swin_40000ep", 
                                             filename="model_final_swin.pth"),
             }
+print("-------------------------check hub: ",hf_hub_download(repo_id="rubikk179/oneformer_cityscapes_Swin_40000ep", 
+                                            filename="model_final_swin.pth"))
 
 CONVNEXT_CFG_DICT = {"cityscapes": "configs/cityscapes/oneformer_convnext_large_bs16_90k.yaml"}