PeiqingYang commited on
Commit
228a5fd
·
1 Parent(s): 6b3879a
Files changed (1) hide show
  1. 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.clear_memory()
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.clear_memory()
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: