Update app.py

app.py

@@ -10,8 +10,8 @@ model = models.densenet121(pretrained=True)
 
 # Modify the classifier layer to output probabilities for 14 classes (pathologies)
 num_classes = 14
-model.fc = nn.Sequential(
-    nn.Linear(model.fc.in_features, num_classes),
+model.classifier = nn.Sequential(
+    nn.Linear(model.classifier.in_features, num_classes),
     nn.Sigmoid(),  # Use Sigmoid for multi-label classification
 )
 
@@ -49,7 +49,7 @@ def predict_disease(image):
 # Gradio Interface
 interface = gr.Interface(
     fn=predict_disease,
-    inputs=gr.inputs.Image(type='pil'),  # Input is an image
+    inputs=gr.components.Image(type='pil'),  # Updated input component
     outputs="label",  # Output is a dictionary of labels with probabilities
     title="CheXNet Pneumonia Detection",
     description="Upload a chest X-ray to detect the probability of 14 different diseases.",
@@ -57,4 +57,4 @@ interface = gr.Interface(
 
 # Launch the Gradio app
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()