Update app.py

app.py

@@ -234,7 +234,6 @@ def pipe_generate(img,p1,p2,motion,time,title):
         guidance_scale=accu,
         num_videos_per_prompt=1,
         num_frames=(fps*time),
-        need_safetycheck=False,
         generator=torch.Generator(device).manual_seed(int(str(random.random()).split(".")[1]))
     ).frames[0]
 
@@ -270,7 +269,10 @@ def handle_generate(*_inp):
         pipe_out.save(name)
     else:
         export_to_video(pipe_out,name,fps=fps)
-    return name
+    if inp[4] == 0:
+        return name, ""
+    else:
+        return "", name
 
 def ui():
     global result
@@ -329,8 +331,10 @@ def ui():
                             value="",
                             interactive=True
                         )
-                with gr.Row():
-                    result = gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=True, type='filepath', show_share_button=False)
+                with gr.Row(scale=0.5):
+                    res_img = gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=True, type='filepath', show_share_button=False)
+                with gr.Row(scale=0.5):
+                    res_vid = gr.Video(interactive=False,elem_classes="image-container", label="Result", show_label=True, type='filepath', show_share_button=False)
         with gr.Row():
             run_button = gr.Button("Start!",elem_classes="btn",scale=0)
 
@@ -342,7 +346,7 @@ def ui():
             ],
             fn=handle_generate,
             inputs=[img,prompt,prompt2,motion,time,title],
-            outputs=result
+            outputs=[res_img,res_vid]
         )
         demo.queue().launch()