lixiang46 commited on
Commit
bee69a0
β€’
1 Parent(s): 211b362
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -92,7 +92,7 @@ def infer(prompt,
92
  scale = 0.8
93
  pipe.set_face_fidelity_scale(scale)
94
 
95
- face_info_generator = FaceInfoGenerator(root_dir = "./")
96
  face_info = face_info_generator.get_faceinfo_one_img(image)
97
  face_bbox_square = face_bbox_to_square(face_info["bbox"])
98
  crop_image = image.crop(face_bbox_square)
 
92
  scale = 0.8
93
  pipe.set_face_fidelity_scale(scale)
94
 
95
+ face_info_generator = FaceInfoGenerator()
96
  face_info = face_info_generator.get_faceinfo_one_img(image)
97
  face_bbox_square = face_bbox_to_square(face_info["bbox"])
98
  crop_image = image.crop(face_bbox_square)