Update gradio_app.py

gradio_app.py

@@ -38,11 +38,11 @@ ConvNeXt_MODEL_DICT = {"cityscapes": hf_hub_download(repo_id="rubikk179/oneforme
                                             filename="model_final_convnext.pth"),
             }
 
-MODEL_DICT = {"DiNAT-L": DINAT_MODEL_DICT,
-        "Swin-L": SWIN_MODEL_DICT }
+MODEL_DICT = {"ConvNeXt": ConvNeXt_MODEL_DICT,
+        "Swin": SWIN_MODEL_DICT }
 
-CFG_DICT = {"DiNAT-L": DINAT_CFG_DICT,
-        "Swin-L": SWIN_CFG_DICT }
+CFG_DICT = {"ConvNeXt": ConvNeXtT_CFG_DICT,
+        "Swin": SWIN_CFG_DICT }
 
 WIDTH_DICT = {"cityscapes": 512}