Update app.py

app.py

@@ -24,8 +24,8 @@ import numpy as np
 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)
+    device = torch.device("cuda") #if torch.cuda.is_available() else torch.device("cpu")
+ 
     parser, parser_up = create_argparser()
     
     args = parser.parse_args()