Update app.py

app.py

@@ -373,7 +373,7 @@ def predict(inp):
         periods_classification = periods_model(image_tensor.unsqueeze(0))
         recon_tensor = reconstruction[0].repeat(3, 1, 1)
         recon_tensor = invTrans(kornia.enhance.invert(recon_tensor))
-        plot_recon = recon_tensor.permute(1, 2, 0).detach().numpy()
+        plot_recon = recon_tensor.permute(1, 2, 0).detach().cpu().numpy()
         w, h = inp.size
         # plot_recon = resize(plot_recon, (h, w))
         m = nn.Sigmoid()