init

app.py

@@ -75,7 +75,7 @@ model_dir = os.path.join(os.path.abspath(os.path.dirname(__file__)), 'U-2-Net/sa
 print("...load U2NET---173.6 MB")
 net = U2NET(3,1)
 
-net.load_state_dict(torch.load(model_dir))
+net.load_state_dict(torch.load(model_dir, map_location='cpu'))
 # if torch.cuda.is_available():
 #     net.cuda()
 net.eval()