root
commited on
Commit
·
33feb37
1
Parent(s):
241c9d1
removing face swap for now
Browse files- handler.py +3 -3
handler.py
CHANGED
|
@@ -366,13 +366,13 @@ class EndpointHandler():
|
|
| 366 |
|
| 367 |
# Perform face swapping
|
| 368 |
# self.print_directory_contents(temp_dir)
|
| 369 |
-
swapped_face_video_path = os.path.join(save_dir, "swapped_face_output.mp4")
|
| 370 |
-
self._swap_face(cropped_face_path, animation_path, swapped_face_video_path)
|
| 371 |
|
| 372 |
# Slow down the produced video by 4x
|
| 373 |
self.print_directory_contents(temp_dir)
|
| 374 |
slowed_down_animation_path = os.path.join(save_dir, "slowed_down_animation_output.mp4")
|
| 375 |
-
self.slow_down_video(
|
| 376 |
|
| 377 |
# Clear CUDA cache before RIFE interpolation
|
| 378 |
torch.cuda.empty_cache()
|
|
|
|
| 366 |
|
| 367 |
# Perform face swapping
|
| 368 |
# self.print_directory_contents(temp_dir)
|
| 369 |
+
# swapped_face_video_path = os.path.join(save_dir, "swapped_face_output.mp4")
|
| 370 |
+
# self._swap_face(cropped_face_path, animation_path, swapped_face_video_path)
|
| 371 |
|
| 372 |
# Slow down the produced video by 4x
|
| 373 |
self.print_directory_contents(temp_dir)
|
| 374 |
slowed_down_animation_path = os.path.join(save_dir, "slowed_down_animation_output.mp4")
|
| 375 |
+
self.slow_down_video(animation_path, slowed_down_animation_path, factor=4)
|
| 376 |
|
| 377 |
# Clear CUDA cache before RIFE interpolation
|
| 378 |
torch.cuda.empty_cache()
|