root
commited on
Commit
·
a7f2e6c
1
Parent(s):
5a9c05f
einops library
Browse files- __pycache__/handler.cpython-310.pyc +0 -0
- handler.py +1 -1
- output/gradio/animation_output.mp4 +0 -0
- output/gradio/cropped_face.jpg +0 -0
- requirements.txt +4 -1
__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.
|
| 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
|