Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+import gradio as 
 import os
 import cv2
 import numpy as np
@@ -98,13 +98,13 @@ def create_video(frames, fps):
     return 'movie.mp4'
 
 
-def infer(prompt,video_in, seed_in, trim_value):
+def infer(prompt, video_in, seed_in, trim_value):
     print(prompt)
     break_vid = get_frames(video_in)
     
-    frames_list= break_vid[0]
+    frames_list = break_vid[0]
     fps = break_vid[1]
-    n_frame = int(trim_value*fps)
+    n_frame = int(trim_value * fps)
     
     if n_frame >= len(frames_list):
         print("video is shorter than the cut value")
@@ -117,11 +117,6 @@ def infer(prompt,video_in, seed_in, trim_value):
         pil_i = Image.open(i).convert("RGB")
         
         pix2pix_img = pix2pix(pil_i, prompt, 50, seed_in, 7.5, 1.5)
-        #print(pix2pix_img)
-        #image = Image.open(pix2pix_img)
-        #rgb_im = image.convert("RGB")
-  
-        # exporting the image
         pix2pix_img.save(f"result_img-{i}.jpg")
         result_frames.append(f"result_img-{i}.jpg")
         print("frame " + i + "/" + str(n_frame) + ": done;")
@@ -184,7 +179,7 @@ with gr.Blocks(css='style.css') as demo:
                 prompt = gr.Textbox(label="Prompt", placeholder="enter prompt", show_label=False, elem_id="prompt-in")
                 with gr.Row():
                     seed_inp = gr.Slider(label="Seed", minimum=0, maximum=2147483647, step=1, value=123456)
-                    trim_in = gr.Slider(label="Cut video at (s)", minimun=1, maximum=5, step=1, value=1)
+                    trim_in = gr.Slider(label="Cut video at (s)", minimum=1, maximum=50, step=1, value=5)  # Updated here
             with gr.Column():
                 video_out = gr.Video(label="Pix2pix video result", elem_id="video-output")
                 gr.HTML("""
@@ -198,25 +193,12 @@ with gr.Blocks(css='style.css') as demo:
                     loading_icon = gr.HTML(loading_icon_html)
                     share_button = gr.Button("Share to community", elem_id="share-btn")
         
-        inputs = [prompt,video_inp,seed_inp, trim_in]
+        inputs = [prompt, video_inp, seed_inp, trim_in]
         outputs = [video_out, share_group]
         
-        #ex = gr.Examples(
-        #    [
-        #        ["Make it a marble sculpture", "./examples/pexels-jill-burrow-7665249_512x512.mp4", 422112651, 4],
-        #        ["Make it molten lava", "./examples/Ocean_Pexels_ 8953474_512x512.mp4", 43571876, 4]
-        #    ],
-        #    inputs=inputs,
-        #    outputs=outputs,
-        #    fn=infer,
-        #    cache_examples=True,
-        #)
-        
         gr.HTML(article)
     
     submit_btn.click(infer, inputs, outputs)
     share_button.click(None, [], [], _js=share_js)
-
-    
     
 demo.queue(max_size=12).launch()