Update app.py

app.py

@@ -17,7 +17,7 @@ def classify_image(image):
     # Make prediction
     with torch.no_grad():
         outputs_pneumonia = model_pneumonia(**inputs_pneumonia)
-        logits_pneumonia = outputs.logits
+        logits_pneumonia = outputs_pneumonia.logits_pneumonia
     # Retrieve the highest probability class label index
     predicted_class_idx_pneumonia = logits_pneumonia.argmax(-1).item()
     # Define a manual mapping of label indices to human-readable labels