Update app.py

app.py

@@ -47,8 +47,8 @@ configs = {
     }
 
 MODEL_REGISTRY = {
-    # "CheXFormer-small": "m42-health/CXformer-small",
-    "CheXFormer-base": "m42-health/CXformer-base",
+    "CheXFormer-small": "m42-health/CXformer-small",
+    # "CheXFormer-base": "m42-health/CXformer-base",
     "ViT-base-16": "google/vit-base-patch16-224",
 }
 
@@ -299,7 +299,7 @@ example_list = [
 gradio_app = gradio.Interface(
     fn     = partial(run_diagnosis, preprocess_fn = preprocess_fxn, Idx2labels = labels_dict),
 
-    inputs = [gradio.Dropdown(["CheXFormer-base", "ViT-base-16"], value="ViT-base-16", label="Select Backbone Model"),
+    inputs = [gradio.Dropdown(["CheXFormer-small", "ViT-base-16"], value="CheXFormer-small", label="Select Backbone Model"),
               gradio.Image(type="pil", label="Load chest-X-ray image here"),
               gradio.Slider(minimum = 0.1, maximum = 0.9, step = 0.05, value = 0.4, label = "Set Prediction Threshold")],