print the torch device

app.py

@@ -8,6 +8,7 @@ import PIL.Image as Image
 
 # Set device to GPU if available
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+print(f"PyTorch Device {device}")
 
 # Load the OpenCLIP model and the necessary preprocessors
 # openclip_model = 'laion/CLIP-ViT-B-32-laion2B-s34B-b79K'