check device

app.py

@@ -63,7 +63,9 @@ if not os.path.exists("./checkpoints/"):
 print("########################Setup done!########################")
 
 # Load U-2-Net model
+print(f"####Using {device}.#####")
 u2net = u2net_load.model(model_name = 'u2netp')
+
 # Main inference function
 def inference(clothing_image, person_image, remove_bg):
     """