root commited on
Commit
a7f2e6c
·
1 Parent(s): 5a9c05f

einops library

Browse files
__pycache__/handler.cpython-310.pyc CHANGED
Binary files a/__pycache__/handler.cpython-310.pyc and b/__pycache__/handler.cpython-310.pyc differ
 
handler.py CHANGED
@@ -67,7 +67,7 @@ class EndpointHandler():
67
  scheduler=scheduler
68
  ).to(device, dtype=self.weight_dtype)
69
 
70
- def _crop_face(self, image, save_path="cropped_face.jpg", margin=0.3):
71
  # Convert image to OpenCV format
72
  cv_image = np.array(image)
73
  cv_image = cv_image[:, :, ::-1].copy()
 
67
  scheduler=scheduler
68
  ).to(device, dtype=self.weight_dtype)
69
 
70
+ def _crop_face(self, image, save_path="cropped_face.jpg", margin=0.5):
71
  # Convert image to OpenCV format
72
  cv_image = np.array(image)
73
  cv_image = cv_image[:, :, ::-1].copy()
output/gradio/animation_output.mp4 CHANGED
Binary files a/output/gradio/animation_output.mp4 and b/output/gradio/animation_output.mp4 differ
 
output/gradio/cropped_face.jpg CHANGED
requirements.txt CHANGED
@@ -32,4 +32,7 @@ omegaconf==2.2.3
32
 
33
  # Face swap related dependencies
34
  facenet-pytorch==2.5.2
35
- dlib==19.22.0
 
 
 
 
32
 
33
  # Face swap related dependencies
34
  facenet-pytorch==2.5.2
35
+ dlib==19.22.0
36
+
37
+ # Stuff huggingface is complaining about
38
+ einops