Spaces:

Santipab
/

Vertebral-Compression-Fracture-Detection

Running

Santipab commited on 9 days ago

Commit

1db8f79

verified ·

1 Parent(s): 6c47a25

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -199,7 +199,7 @@ elif feature == "AP - Detection":
         st.subheader("3️⃣ Heatmap")
     args = Namespace(
-        resume="model_50.pth",
         data_dir=os.path.join(REPO, "dataPath"),
         dataset="spinal",
         phase="test",

         st.subheader("3️⃣ Heatmap")
     args = Namespace(
+        resume="model_30.pth",
         data_dir=os.path.join(REPO, "dataPath"),
         dataset="spinal",
         phase="test",