Update gradio_app.py

gradio_app.py

@@ -104,7 +104,7 @@ def infer(frame1_path, frame2_path):
         weighted_average=weighted_average, # True
         noise_injection_steps=noise_injection_steps, # 0
         noise_injection_ratio= noise_injection_ratio, # 0.5
-         decode_chunk_size=4
+         decode_chunk_size=8
     ).frames[0]
 
     print(f"FRAMES: {frames}")
@@ -113,15 +113,15 @@ def infer(frame1_path, frame2_path):
 
     check_outputs_folder(out_dir)
     os.makedirs(out_dir, exist_ok=True)
-    out_path = "result/video_result.gif"
+    out_path = "result/video_result.mp4"
 
-    '''
+    
     if out_path.endswith('.gif'):
         frames[0].save(out_path, save_all=True, append_images=frames[1:], duration=142, loop=0)
     else:
         export_to_video(frames, out_path, fps=7)
-    '''
-    return "done"
+    
+    return out_path
 
 with gr.Blocks() as demo:
 
@@ -133,7 +133,7 @@ with gr.Blocks() as demo:
                 image_input2 = gr.Image(type="filepath")
                 submit_btn = gr.Button("Submit")
             with gr.Column():
-                output = gr.Textbox()
+                output = gr.Video()
     
     submit_btn.click(
         fn = infer,