Update app.py

app.py

@@ -25,6 +25,7 @@ from huggingface_hub import hf_hub_download
 
 def run(image, mode, sample_c=1.3,  num_samples=3, sample_step=100):
     device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
+    print(device)
     parser, parser_up = create_argparser()
     
     args = parser.parse_args()