fffiloni commited on
Commit
75046ec
·
verified ·
1 Parent(s): 47a3fd9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -8,7 +8,7 @@ from moviepy.editor import *
8
  from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
9
  from gradio_client import Client
10
  import gradio as gr
11
- from share_btn import community_icon_html, loading_icon_html, share_js
12
 
13
  matte_client = Client("https://fffiloni-video-matting-anything.hf.space/")
14
 
@@ -109,7 +109,7 @@ def infer(video_frames, masks_frames, project_name):
109
  for item in results_folder_content:
110
  print(item)
111
 
112
- return results_folder_content[0], results_folder_content[1], gr.Group.update(visible=True)
113
 
114
 
115
  def get_frames(video_in, img_type):
@@ -321,7 +321,7 @@ def infer_auto(project_name, video_in, subject_to_remove):
321
  for item in results_folder_content:
322
  print(item)
323
 
324
- return results_folder_content[0], results_folder_content[1], gr.Group.update(visible=True)
325
 
326
  css="""
327
  #col-container{
@@ -475,8 +475,8 @@ with gr.Blocks(css=css) as demo:
475
 
476
 
477
 
478
- submit_btn.click(fn=infer, inputs=[video_frames, masks_frames, project_name], outputs=[res_masked, res_files, share_group])
479
- submit_auto_btn.click(fn=infer_auto, inputs=[project_name_2, video_in, subject_to_remove], outputs=[res_masked, res_files, share_group])
480
 
481
 
482
  demo.queue(max_size=12).launch()
 
8
  from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
9
  from gradio_client import Client
10
  import gradio as gr
11
+
12
 
13
  matte_client = Client("https://fffiloni-video-matting-anything.hf.space/")
14
 
 
109
  for item in results_folder_content:
110
  print(item)
111
 
112
+ return results_folder_content[0], results_folder_content[1]
113
 
114
 
115
  def get_frames(video_in, img_type):
 
321
  for item in results_folder_content:
322
  print(item)
323
 
324
+ return results_folder_content[0], results_folder_content[1]
325
 
326
  css="""
327
  #col-container{
 
475
 
476
 
477
 
478
+ submit_btn.click(fn=infer, inputs=[video_frames, masks_frames, project_name], outputs=[res_masked, res_files])
479
+ submit_auto_btn.click(fn=infer_auto, inputs=[project_name_2, video_in, subject_to_remove], outputs=[res_masked, res_files])
480
 
481
 
482
  demo.queue(max_size=12).launch()