Update app.py

app.py

@@ -23,7 +23,7 @@ conditions = [
 # Load and configure the model
 model = models.densenet121(pretrained=True)  # DenseNet pre-trained on ImageNet
 num_features = model.classifier.in_features
-model.classifier = torch.nn.Linear(num_features, len(conditions))  # Output for all 16 conditions
+model.classifier = torch.nn.Linear(num_features, len(conditions))  # Output for all 24 conditions
 model.eval()
 
 # Define device