cfoli commited on
Commit
edab602
·
1 Parent(s): 66210cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -295,7 +295,7 @@ example_list = [
295
  gradio_app = gradio.Interface(
296
  fn = partial(run_diagnosis, preprocess_fn = preprocess_fxn, Idx2labels = labels_dict, threshold = configs["THRESHOLD"]),
297
 
298
- inputs = [gradio.Dropdown(["ConvNeXt(small)", "ConvNeXt(tiny)", "EfficientNet(v2_small)", "EfficientNet(b3)", "RegNet(x3_2GF)","ResNet50"], value="EfficientNet(b3)", label="Select Backbone Model"),
299
  gradio.Image(type="pil", label="Load chest-X-ray image here")],
300
 
301
  outputs = [gradio.Textbox(label="Predicted Medical Conditions"),
 
295
  gradio_app = gradio.Interface(
296
  fn = partial(run_diagnosis, preprocess_fn = preprocess_fxn, Idx2labels = labels_dict, threshold = configs["THRESHOLD"]),
297
 
298
+ inputs = [gradio.Dropdown(list(BACKBONE_REGISTRY.keys()), value="ResNet50", label="Select Backbone Model"),
299
  gradio.Image(type="pil", label="Load chest-X-ray image here")],
300
 
301
  outputs = [gradio.Textbox(label="Predicted Medical Conditions"),