updated running devices status

app.py

@@ -48,7 +48,10 @@ model = DenseNet121_CheXpert(num_labels=14, pretrained=None)
 model.load_state_dict(state, strict=False)
 model.to(device)
 model.eval()
-print(f"Model loaded successfully on {device}")
+if device.type=='cuda':
+    print(f"Model loaded successfully on GPU {torch.cuda.get_device_name(torch.cuda.current_device())}")
+else:
+    print(f"Model loaded successfully on CPU")
 
 def predict(image, threshold):
     """Generate predictions and Grad-CAM visualizations"""