Update app.py

app.py

@@ -5,6 +5,9 @@ import joblib
 
 def predict(input_image):
     device = torch.device("cpu")
+    # when running on GPUs: 
+    # device = torch.device("cuda")
+    
     model, _, preprocess = open_clip.create_model_and_transforms('ViT-L-14', pretrained='openai', device=device)
 
     image = preprocess(input_image).unsqueeze(0).to(device)