Update app.py

app.py

@@ -152,8 +152,9 @@ ckptspider = torch.load('Spiderverse-face-500iters-8face.pt', map_location=lambd
 generatorspider.load_state_dict(ckptspider["g"], strict=False)
 
 
-def inference(img, model):    
-    aligned_face = align_face(img)
+def inference(img, model):  
+    img.save('out.jpg')  
+    aligned_face = align_face('out.jpg')
         
     my_w = projection(aligned_face, "test.pt", device).unsqueeze(0)
     if model == 'JoJo':