Spaces:
Running
on
L4
Running
on
L4
Commit
·
228a5fd
1
Parent(s):
6b3879a
debug
Browse files- hugging_face/app.py +3 -4
hugging_face/app.py
CHANGED
@@ -116,9 +116,8 @@ def get_frames_from_video(video_input, video_state):
|
|
116 |
user_name = time.time()
|
117 |
|
118 |
# extract Audio
|
119 |
-
audio_path = "audio.wav"
|
120 |
-
audio_path = video_input.replace(".mp4", "_audio.wav")
|
121 |
try:
|
|
|
122 |
ffmpeg.input(video_path).output(audio_path, format='wav', acodec='pcm_s16le', ac=2, ar='44100').run(overwrite_output=True, quiet=True)
|
123 |
except Exception as e:
|
124 |
print(f"Audio extraction error: {str(e)}")
|
@@ -259,7 +258,7 @@ def show_mask(video_state, interactive_state, mask_dropdown):
|
|
259 |
|
260 |
# image matting
|
261 |
def image_matting(video_state, interactive_state, mask_dropdown, erode_kernel_size, dilate_kernel_size, refine_iter):
|
262 |
-
matanyone_processor.
|
263 |
if interactive_state["track_end_number"]:
|
264 |
following_frames = video_state["origin_images"][video_state["select_frame_number"]:interactive_state["track_end_number"]]
|
265 |
else:
|
@@ -287,7 +286,7 @@ def image_matting(video_state, interactive_state, mask_dropdown, erode_kernel_si
|
|
287 |
|
288 |
# video matting
|
289 |
def video_matting(video_state, interactive_state, mask_dropdown, erode_kernel_size, dilate_kernel_size):
|
290 |
-
matanyone_processor.
|
291 |
if interactive_state["track_end_number"]:
|
292 |
following_frames = video_state["origin_images"][video_state["select_frame_number"]:interactive_state["track_end_number"]]
|
293 |
else:
|
|
|
116 |
user_name = time.time()
|
117 |
|
118 |
# extract Audio
|
|
|
|
|
119 |
try:
|
120 |
+
audio_path = video_input.replace(".mp4", "_audio.wav")
|
121 |
ffmpeg.input(video_path).output(audio_path, format='wav', acodec='pcm_s16le', ac=2, ar='44100').run(overwrite_output=True, quiet=True)
|
122 |
except Exception as e:
|
123 |
print(f"Audio extraction error: {str(e)}")
|
|
|
258 |
|
259 |
# image matting
|
260 |
def image_matting(video_state, interactive_state, mask_dropdown, erode_kernel_size, dilate_kernel_size, refine_iter):
|
261 |
+
matanyone_processor = InferenceCore(matanyone_model, cfg=matanyone_model.cfg)
|
262 |
if interactive_state["track_end_number"]:
|
263 |
following_frames = video_state["origin_images"][video_state["select_frame_number"]:interactive_state["track_end_number"]]
|
264 |
else:
|
|
|
286 |
|
287 |
# video matting
|
288 |
def video_matting(video_state, interactive_state, mask_dropdown, erode_kernel_size, dilate_kernel_size):
|
289 |
+
matanyone_processor = InferenceCore(matanyone_model, cfg=matanyone_model.cfg)
|
290 |
if interactive_state["track_end_number"]:
|
291 |
following_frames = video_state["origin_images"][video_state["select_frame_number"]:interactive_state["track_end_number"]]
|
292 |
else:
|