Update app.py
Browse files
app.py
CHANGED
|
@@ -9,8 +9,8 @@ st.set_page_config(page_title="X-ray Diagnosis Demo", layout="centered")
|
|
| 9 |
st.title("🩻 X-ray Multi-Label Diagnosis App (CheXNet)")
|
| 10 |
|
| 11 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 12 |
-
|
| 13 |
-
|
| 14 |
|
| 15 |
uploaded_file = st.file_uploader("Upload a chest X-ray", type=["jpg", "jpeg", "png"])
|
| 16 |
|
|
|
|
| 9 |
st.title("🩻 X-ray Multi-Label Diagnosis App (CheXNet)")
|
| 10 |
|
| 11 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 12 |
+
model = load_model(device)
|
| 13 |
+
|
| 14 |
|
| 15 |
uploaded_file = st.file_uploader("Upload a chest X-ray", type=["jpg", "jpeg", "png"])
|
| 16 |
|