update code

app.py

@@ -26,7 +26,7 @@ def predict(img):
     with torch.no_grad():
         img=align(img)
         image_tensor=TF.to_tensor(img).cpu()
-        image_tensor=image_tensor.unsqueeze(0).cuda()
+        image_tensor=image_tensor.unsqueeze(0).cpu()
         ipt=net_c(image_tensor)
         image_tensor = image_tensor.half()
         ipt = ipt.half()