Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -43,7 +43,7 @@ configs = {
|
|
| 43 |
|
| 44 |
"DEFAULT_BACKBONE": "CheXFormer-small",
|
| 45 |
|
| 46 |
-
"THRESHOLD": 0.
|
| 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.
|
| 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)],
|