switched device to cuda

app.py

@@ -166,7 +166,7 @@ args = parser.parse_args()
 #    args.device = torch.device('cuda')
 #else:
 #    args.device = torch.device('cpu')
-args.device = torch.device('cpu')
+args.device = torch.device('cuda')
 model, transform = build_model_and_transforms(args)
 
 examples = [
@@ -315,6 +315,7 @@ def count(image, text, prompts, state, device):
 @spaces.GPU
 def count_main(image, text, prompts, device):
     model.to(device)
+    print("model device: " + str(model.device))
     keywords = "" # do not handle this for now
     # Handle no prompt case.
     if prompts is None: