completed demo

florence_sam/app.py

@@ -4,17 +4,13 @@ import gradio as gr
 import os
 
 def pre_processor(video_path, scale_factor, prompt, crop_duration):
-    # Load video with moviepy
     video = mp.VideoFileClip(video_path)
 
-    # Crop video to the specified duration (in seconds)
     cropped_video = video.subclip(0, min(crop_duration, video.duration))
 
-    # Save the cropped video to a temporary file
     temp_output = "cropped_video.mp4"
     cropped_video.write_videofile(temp_output, codec="libx264")
 
-    # Pass the cropped video and other parameters to the infer function
     output = infer(temp_output, scale_factor, prompt)
 
     # Clean up temporary files
@@ -25,6 +21,7 @@ def pre_processor(video_path, scale_factor, prompt, crop_duration):
 
 
 demo = gr.Interface(
+    title="Text based video inpainting",
     fn=pre_processor, 
     inputs=[
         gr.Video(label="Upload Video"), 

florence_sam/main.py

@@ -30,6 +30,7 @@ def infer(video_path, scale_factor, prompt):
 
     # remove intermediate files
     shutil.rmtree("tmp")
+    torch.cuda.empty_cache()
 
     return "results/input_frames/inpaint_out.mp4"