Indicate if cuda was found

app.py

@@ -34,6 +34,7 @@ def main():
     if torch.cuda.is_available():
         device = "cuda:0"
         torch.cuda.set_device(device)
+        print("Found cuda")
     else:
         device = "cpu"