add gpu support

app.py

@@ -43,7 +43,7 @@ def count_barnacles(input_img, progress=gr.Progress()):
     predicted_labels_list = []
     for data in progress.tqdm(test_dataloader):
         with torch.no_grad():
-            data.to("cuda")
+            data = data.to("cuda")
             predicted_labels_list += [model(data)]
     predicted_labels = torch.cat(predicted_labels_list)
     x = int(math.sqrt(predicted_labels.shape[0]))