Ahsen Khaliq commited on
Commit
76c051f
1 Parent(s): 8b17f8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -108,8 +108,8 @@ def inference(img, model):
108
 
109
  face_helper.read_image(img)
110
  face_helper.get_face_landmarks_5(only_center_face=False, eye_dist_threshold=5)
111
- face_helper.align_warp_face(save_cropped_path=".")
112
- pilimg = Image.open("./_00.png")
113
 
114
  my_w = e4e_projection(pilimg, "test.pt", device).unsqueeze(0)
115
  if model == 'JoJo':
 
108
 
109
  face_helper.read_image(img)
110
  face_helper.get_face_landmarks_5(only_center_face=False, eye_dist_threshold=5)
111
+ face_helper.align_warp_face(save_cropped_path="/home/user/app/")
112
+ pilimg = Image.open("/home/user/app/_02.png")
113
 
114
  my_w = e4e_projection(pilimg, "test.pt", device).unsqueeze(0)
115
  if model == 'JoJo':