Check device

app.py

@@ -21,6 +21,7 @@ parser.add_argument('--ip', type=str, default=None)
 args = parser.parse_args()
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+print(f'Device: {device}')
 os.makedirs("results", exist_ok=True)
 os.makedirs("models", exist_ok=True)