fix

app.py

@@ -67,7 +67,7 @@ def inference(img):
         
     my_w = projection(aligned_face, "test.pt", device).unsqueeze(0)           
     
-    npimage = my_w.permute(1, 2, 0).detach().numpy()
+    npimage = my_w.numpy()
     imageio.imwrite('filename.jpeg', npimage)
     return 'filename.jpeg'