Update app.py

app.py

@@ -56,6 +56,8 @@ def generate(input_video):
             check=True
         )
 
+        print(output_videos)
+
         subprocess.run(
             [
                 "python", "infer_4d.py", "big",
@@ -70,7 +72,7 @@ def generate(input_video):
 
         output_videos = glob(os.path.join(f"{workdir}", "*.mp4"))
         print(output_videos)
-        return output_videos[0]
+        return output_videos[0], output_videos[1], output_videos[2], output_videos[3],output_videos[4] 
    
     except subprocess.CalledProcessError as e:
         raise gr.Error(f"Error during inference: {str(e)}")
@@ -82,7 +84,11 @@ with gr.Blocks() as demo:
                 input_video = gr.Video(label="Input Video")
                 submit_btn = gr.Button("Submit")
             with gr.Column():
-                output_result = gr.Video(label="Result")
+                output_result_0 = gr.Video(label="Result")
+                output_result_1 = gr.Video(label="Result")
+                output_result_2 = gr.Video(label="Result")
+                output_result_3 = gr.Video(label="Result")
+                output_result_4 = gr.Video(label="Result")
 
         gr.Examples(
             examples = video_examples,
@@ -92,7 +98,13 @@ with gr.Blocks() as demo:
     submit_btn.click(
         fn = generate,
         inputs = [input_video],
-        outputs = [output_result]
+        outputs = [
+            output_result_0, 
+            output_result_1,
+            output_result_2,
+            output_result_3,
+            output_result_4
+        ]
     )
 
 demo.queue().launch(show_api=False, show_error=True)