Hugging Face's logo

Spaces:
diegoccx
/
OneFormerMentoria
Running

App Files Community
Jitesh Jain commited on
Commit
d3c3aa9
·
1 Parent(s): 378b71d

Update model repos names

Browse files
Files changed (1) hide show
  1. gradio_app.py +6 -6
gradio_app.py CHANGED
@@ -32,11 +32,11 @@ SWIN_CFG_DICT = {"cityscapes": "configs/cityscapes/oneformer_swin_large_IN21k_38
32
  "coco": "configs/coco/oneformer_swin_large_IN21k_384_bs16_100ep.yaml",
33
  "ade20k": "configs/ade20k/oneformer_swin_large_IN21k_384_bs16_160k.yaml",}
34
 
35
- SWIN_MODEL_DICT = {"cityscapes": hf_hub_download(repo_id="shi-labs/swin_l_oneformer_cityscapes",
36
  filename="250_16_swin_l_oneformer_cityscapes_90k.pth"),
37
- "coco": hf_hub_download(repo_id="shi-labs/swin_l_oneformer_coco",
38
  filename="150_16_swin_l_oneformer_coco_100ep.pth"),
39
- "ade20k": hf_hub_download(repo_id="shi-labs/swin_l_oneformer_ade20k",
40
  filename="250_16_swin_l_oneformer_ade20k_160k.pth")
41
  }
42
 
@@ -44,11 +44,11 @@ DINAT_CFG_DICT = {"cityscapes": "configs/cityscapes/oneformer_dinat_large_bs16_9
44
  "coco": "configs/coco/oneformer_dinat_large_bs16_100ep.yaml",
45
  "ade20k": "configs/ade20k/oneformer_dinat_large_IN21k_384_bs16_160k.yaml",}
46
 
47
- DINAT_MODEL_DICT = {"cityscapes": hf_hub_download(repo_id="shi-labs/dinat_l_oneformer_cityscapes",
48
  filename="250_16_dinat_l_oneformer_cityscapes_90k.pth"),
49
- "coco": hf_hub_download(repo_id="shi-labs/dinat_l_oneformer_coco",
50
  filename="150_16_dinat_l_oneformer_coco_100ep.pth"),
51
- "ade20k": hf_hub_download(repo_id="shi-labs/dinat_l_oneformer_ade20k",
52
  filename="250_16_dinat_l_oneformer_ade20k_160k.pth")
53
  }
54
 
 
32
  "coco": "configs/coco/oneformer_swin_large_IN21k_384_bs16_100ep.yaml",
33
  "ade20k": "configs/ade20k/oneformer_swin_large_IN21k_384_bs16_160k.yaml",}
34
 
35
+ SWIN_MODEL_DICT = {"cityscapes": hf_hub_download(repo_id="shi-labs/oneformer_cityscapes_swin_large",
36
  filename="250_16_swin_l_oneformer_cityscapes_90k.pth"),
37
+ "coco": hf_hub_download(repo_id="shi-labs/oneformer_coco_swin_large",
38
  filename="150_16_swin_l_oneformer_coco_100ep.pth"),
39
+ "ade20k": hf_hub_download(repo_id="shi-labs/oneformer_ade20k_swin_large",
40
  filename="250_16_swin_l_oneformer_ade20k_160k.pth")
41
  }
42
 
 
44
  "coco": "configs/coco/oneformer_dinat_large_bs16_100ep.yaml",
45
  "ade20k": "configs/ade20k/oneformer_dinat_large_IN21k_384_bs16_160k.yaml",}
46
 
47
+ DINAT_MODEL_DICT = {"cityscapes": hf_hub_download(repo_id="shi-labs/oneformer_cityscapes_dinat_large",
48
  filename="250_16_dinat_l_oneformer_cityscapes_90k.pth"),
49
+ "coco": hf_hub_download(repo_id="shi-labs/oneformer_coco_dinat_large",
50
  filename="150_16_dinat_l_oneformer_coco_100ep.pth"),
51
+ "ade20k": hf_hub_download(repo_id="shi-labs/oneformer_ade20k_dinat_large",
52
  filename="250_16_dinat_l_oneformer_ade20k_160k.pth")
53
  }
54