cfgpp commited on
Commit
3367ea9
·
verified ·
1 Parent(s): 348a42c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- model_path = "model/dannynet-55-best_model_20250422-211522.pth"
13
- model = load_model(model_path, device)
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