Spaces:

luxmorocco
/

Doctoria_CXR

Sleeping

luxmorocco commited on Jan 27, 2024

Commit

d67965f

•

1 Parent(s): 5587829

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -141,7 +141,7 @@ uploaded_file = st.sidebar.file_uploader("Upload Chest X-ray image", type=["png"
 # Load the model (use your model loading function)
 # Ensure the model path is correct and accessible
 model = create_model(num_classes=14)
-model.load_state_dict(torch.load('Doctoria_CXR/Doctoria CXR Thoraric Full Model.pth', map_location=torch.device('cpu')))
 model.eval()
 def process_image(image_path):

 # Load the model (use your model loading function)
 # Ensure the model path is correct and accessible
 model = create_model(num_classes=14)
+model.load_state_dict(torch.load('Doctoria CXR Thoraric Full Model.pth', map_location=torch.device('cpu')))
 model.eval()
 def process_image(image_path):