Update app.py

app.py

@@ -42,7 +42,7 @@ configs = {
     "MEAN": (0.485, 0.456, 0.406),
     "STD":  (0.229, 0.224, 0.225),
 
-    "DEFAULT_BACKBONE": "EfficientNet(b3)",
+    "DEFAULT_BACKBONE": "ConvNeXt(tiny)",
     "THRESHOLD": 0.5
     }
 
@@ -289,7 +289,7 @@ example_list_img_names = os.listdir(example_list_dir)
 
 example_list = [
     [configs["DEFAULT_BACKBONE"], os.path.join(example_list_dir, example_img)]
-    for example_img in example_list_img_names[:6]
+    for example_img in example_list_img_names[:8]
     if example_img.lower().endswith(".png")]
 
 # example_list = [['/content/new_labels.csv',"ResNet50"]]
@@ -299,7 +299,7 @@ gradio_app = gradio.Interface(
 
     inputs = [gradio.Dropdown(list(BACKBONE_REGISTRY.keys()), value="ResNet50", 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.1, value = 0.4, label = "Set Prediction Threshold")
+              gradio.Slider(minimum = 0.1, maximum = 0.9, step = 0.05, value = 0.4, label = "Set Prediction Threshold")
              ],
 
     outputs = [gradio.Textbox(label="Predicted Medical Conditions"),