Update app.py

app.py

@@ -147,7 +147,7 @@ def generate_video_from_text(text, opt, pipeline):
     motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
     
     save_dir = 'temp/'
-    filename = generate_md5(str(time.time())) + ".mp4"
+    filename = generate_md5(str(time.time())) + ".gif"
     save_path = pjoin(save_dir, str(filename))
     os.makedirs(save_dir, exist_ok=True)
     
@@ -195,7 +195,7 @@ def generate_video_from_text(text, opt, pipeline):
     exc = end_time - start_time
     gr.Info(f"Rendering time cost: {exc:.2f} s", duration = 3)
     
-    video_dis = f'<video controls playsinline width="{width}" style="display: block; margin: 0 auto;" src="/gradio_api/file={save_path}"></video>'
+    video_dis = f'<img src="/gradio_api/file={save_path}" width="{width}" style="display: block; margin: 0 auto;">'
     style_dis = video_dis + """<br> <p align="center"> Content Reference </p>"""
     global edit_config
     edit_config = set_all_use_to_false(edit_config)
@@ -231,7 +231,7 @@ def reweighting(text, idx, weight, opt, pipeline):
     motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
     
     save_dir = 'temp/'
-    filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
+    filenames = [generate_md5(str(time.time())) + ".gif", generate_md5(str(time.time())) + ".gif"]
     save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1]))]
     os.makedirs(save_dir, exist_ok=True)
     
@@ -274,7 +274,7 @@ def reweighting(text, idx, weight, opt, pipeline):
     exc = end_time - start_time
     gr.Info(f"Rendering time cost: {exc:.2f} s", duration = 3)
     
-    video_dis = f'<video controls playsinline width="{width}" style="display: block; margin: 0 auto;" src="/gradio_api/file={save_paths[1]}"></video>'
+    video_dis = f'<img width="{width}" style="display: block; margin: 0 auto;" src="/gradio_api/file={save_paths[1]}">'
     
     
     edit_config = set_all_use_to_false(edit_config)
@@ -310,7 +310,7 @@ def generate_example_based_motion(text, chunk_size, example_based_steps_end, tem
     motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
     
     save_dir = 'temp/'
-    filenames = [generate_md5(str(time.time())) + ".mp4" for _ in range(num_motion)]
+    filenames = [generate_md5(str(time.time())) + ".gif" for _ in range(num_motion)]
     save_paths = [pjoin(save_dir, str(filenames[i])) for i in range(num_motion)]
     os.makedirs(save_dir, exist_ok=True)
     
@@ -352,7 +352,7 @@ def generate_example_based_motion(text, chunk_size, example_based_steps_end, tem
         kinematic_tree = paramUtil.t2m_kinematic_chain if (opt.dataset_name == 't2m') else paramUtil.kit_kinematic_chain
         plot_3d_motion(save_paths[i], kinematic_tree, motion, title=title, fps=opt.fps, radius=opt.radius)
         video_html = f'''
-        <video class="retrieved_video" width="{width}" height="{height}" preload="auto" muted playsinline onpause="this.load()" autoplay loop disablepictureinpicture src="/gradio_api/file=={save_paths[i]}"> </video>
+        <img class="retrieved_video" width="{width}" height="{height}" preload="auto" src="/gradio_api/file={save_paths[i]}">
         '''
         video_dis.append(video_html)
         i += 1
@@ -395,7 +395,7 @@ def transfer_style(text, style_text, style_transfer_steps_end, opt, pipeline):
     motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
     
     save_dir = 'temp/'
-    filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
+    filenames = [generate_md5(str(time.time())) + ".gif", generate_md5(str(time.time())) + ".gif", generate_md5(str(time.time())) + ".gif"]
     save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1])), pjoin(save_dir, str(filenames[2]))]
     os.makedirs(save_dir, exist_ok=True)
     
@@ -436,10 +436,10 @@ def transfer_style(text, style_text, style_transfer_steps_end, opt, pipeline):
     exc = end_time - start_time
     gr.Info(f"Rendering time cost: {exc:.2f} s", duration = 3)
     
-    video_dis0 = f"""<video controls playsinline width="{width}" style="display: block; margin: 0 auto;" src="/gradio_api/file={save_paths[0]}"></video> <br> <p align="center"> Style Reference </p>"""
-    video_dis1 = f"""<video controls playsinline width="{width}" style="display: block; margin: 0 auto;" src="/gradio_api/file={save_paths[2]}"></video> <br> <p align="center"> Content Reference </p>"""
-    video_dis2 = f"""<video controls playsinline width="{width}" style="display: block; margin: 0 auto;" src="/gradio_api/file={save_paths[1]}"></video> <br> <p align="center"> Transfered Result </p>"""
-    
+    video_dis0 = f"""<img width="{width}" style="display: block; margin: 0 auto;" src="/gradio_api/file={save_paths[0]}"> <br> <p align="center"> Style Reference </p>"""
+    video_dis1 = f"""<img width="{width}" style="display: block; margin: 0 auto;" src="/gradio_api/file={save_paths[2]}"> <br> <p align="center"> Content Reference </p>"""
+    video_dis2 = f"""<img width="{width}" style="display: block; margin: 0 auto;" src="/gradio_api/file={save_paths[1]}"> <br> <p align="center"> Transfered Result </p>"""
+     
     edit_config = set_all_use_to_false(edit_config)
     return video_dis0, video_dis2