NeuralFalcon commited on
Commit
b8c1a34
·
verified ·
1 Parent(s): f95c4ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -254,7 +254,7 @@ def video_watermark_remover(video_path):
254
  global gpu
255
  upload_path=upload_file(video_path)
256
  extract_frames(upload_path, "./frames")
257
- video_path = "/content/face.mp4"
258
  vido_chunks(upload_path)
259
  marge_video(gpu=gpu)
260
  save_path=recover_audio(upload_path)
@@ -265,12 +265,13 @@ import gradio as gr
265
  import click
266
 
267
  def gradio_interface(video_file):
268
- return video_watermark_remover(video_file)
 
269
 
270
  demo = gr.Interface(
271
  fn=gradio_interface,
272
  inputs=gr.File(label="Upload Video"),
273
- outputs=gr.File(label="Processed Video"),
274
  title="Video Watermark Remover",
275
  description="Upload a video, and this tool will remove watermarks using blurring techniques."
276
  )
 
254
  global gpu
255
  upload_path=upload_file(video_path)
256
  extract_frames(upload_path, "./frames")
257
+ batch_process(batch_size=100)
258
  vido_chunks(upload_path)
259
  marge_video(gpu=gpu)
260
  save_path=recover_audio(upload_path)
 
265
  import click
266
 
267
  def gradio_interface(video_file):
268
+ vid_path=video_watermark_remover(video_file)
269
+ return vid_path,vid_path
270
 
271
  demo = gr.Interface(
272
  fn=gradio_interface,
273
  inputs=gr.File(label="Upload Video"),
274
+ outputs=[gr.File(label="Download Video"),gr.Video(label="Play Video")],
275
  title="Video Watermark Remover",
276
  description="Upload a video, and this tool will remove watermarks using blurring techniques."
277
  )