Update app.py

app.py

@@ -208,21 +208,9 @@ 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=15)
-#model.load_state_dict(torch.load('Doctoria CXR Thoraric Full Model.pth', map_location=torch.device('cpu')))
-#model.eval()
-
-def load_model(model_path):
-    # Create an instance of the VinDetector model
-    model = VinDetector(num_classes=14)  # Adjust num_classes as needed
-    
-    # Load the saved state_dict
-    model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
-    
-    model.eval()  # Set the model to evaluation mode
-    return model
-
-model = load_model('Doctoria CXR Thoraric Full Model.pth')
+model = create_model(num_classes=15)
+model.load_state_dict(torch.load('Doctoria CXR Thoraric Full Model.pth', map_location=torch.device('cpu')))
+model.eval()
 
 def process_image(image_path):
     # Load and transform the image