Update app.py

app.py

@@ -35,6 +35,7 @@ app.add_middleware(
 # Load CLIP model from local folder
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print('Device: ', device)
+print("CUDA version: ", torch.version.cuda)
 model, preprocess = clip.load("ViT-B/32", device=device)
 print('Model Params Device: ',next(model.parameters()).device)  # Should print 'cuda:0'