Update app.py

app.py

@@ -34,7 +34,7 @@ def create_video(frames, fps):
     return 'movie.mp4'
 
 
-def infer(prompt,video_in):
+def infer(prompt,video_in, seed_in):
     break_vid = get_frames(video_in)
     
     frames_list= break_vid[0]
@@ -42,7 +42,7 @@ def infer(prompt,video_in):
     result_frames = []
     
     for i in frames_list:
-        pix2pix_img = pix2pix(prompt,5.5,1.5,i,15,"",512,512,123456,fn_index=0)
+        pix2pix_img = pix2pix(prompt,5.5,1.5,i,15,"",512,512,seed_in,fn_index=0)
         images = [os.path.join(pix2pix_img[0], img) for img in os.listdir(pix2pix_img[0])]
         result_frames.append(images[0])
 
@@ -53,16 +53,16 @@ def infer(prompt,video_in):
 
 with gr.Blocks(css='style.css') as demo:
     with gr.Column(elem_id="col-container"):
-        prompt = gr.Textbox(placeholder="enter prompt")
-        video_inp = gr.Video(source="upload", type="filepath")
-        video_out = gr.Video()
-       
-        submit_btn = gr.Button("go")
+        with gr.Row():
+            with gr.Column():
+                prompt = gr.Textbox(placeholder="enter prompt")
+                video_inp = gr.Video(source="upload", type="filepath")
+                seed_inp = Number(value=123456)
+            with gr.Column():
+                video_out = gr.Video()
+                submit_btn = gr.Button("Generate Pix2Pix video")
 
-    inputs = [
-        prompt,
-        video_inp
-    ]
+    inputs = [prompt,video_inp,seed_inp]
     outputs = [video_out]
     
     submit_btn.click(infer, inputs, outputs)