cfoli commited on
Commit
e2def54
·
verified ·
1 Parent(s): c01165b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -43,7 +43,7 @@ configs = {
43
 
44
  "DEFAULT_BACKBONE": "CheXFormer-small",
45
 
46
- "THRESHOLD": 0.5
47
  }
48
 
49
  MODEL_REGISTRY = {
@@ -301,7 +301,7 @@ gradio_app = gradio.Interface(
301
 
302
  inputs = [gradio.Dropdown(["CheXFormer-small", "ViT-base-16"], value="CheXFormer-small", label="Select Backbone Model"),
303
  gradio.Image(type="pil", label="Load chest-X-ray image here"),
304
- gradio.Slider(minimum = 0.1, maximum = 0.9, step = 0.05, value = 0.4, label = "Set Prediction Threshold")],
305
 
306
  outputs = [gradio.Textbox(label="Predicted Medical Condition(s)"),
307
  gradio.Label(label="Predicted Probabilities", show_label=False)],
 
43
 
44
  "DEFAULT_BACKBONE": "CheXFormer-small",
45
 
46
+ "THRESHOLD": 0.2
47
  }
48
 
49
  MODEL_REGISTRY = {
 
301
 
302
  inputs = [gradio.Dropdown(["CheXFormer-small", "ViT-base-16"], value="CheXFormer-small", label="Select Backbone Model"),
303
  gradio.Image(type="pil", label="Load chest-X-ray image here"),
304
+ gradio.Slider(minimum = 0.1, maximum = 0.9, step = 0.05, value = 0.2, label = "Set Prediction Threshold")],
305
 
306
  outputs = [gradio.Textbox(label="Predicted Medical Condition(s)"),
307
  gradio.Label(label="Predicted Probabilities", show_label=False)],