hugging face got too stressed back to video

.ipynb_checkpoints/app-checkpoint.py

@@ -486,7 +486,7 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animate=True):
         
     if animate:
         anim = animation.FuncAnimation(fig, partial(update), frames=100, interval=30)
-        anim.save("prediction.gif")
+        anim.save("prediction.mp4")
         
     return plt, Rm, Ra, Emax, Emin, Vd, Tc, start_v
 
@@ -770,9 +770,9 @@ def generate_example():
         video = f"EchoNet-Dynamic/Videos/{filename}"
 
         plot, Rm, Ra, Emax, Emin, Vd,Tc, start_v = pvloop_simulator(Rm=round(results[4].item(),2), Ra=round(results[5].item(),2), Emax=round(results[2].item(),2), Emin=round(results[3].item(),2), Vd=round(results[6].item(),2), Tc=round(results[0].item(),2), start_v=round(results[1].item(),2))
-        video_file = video.replace("avi", "mp4")
-        video = "<video autoplay loop muted> <source src=video_file type='video/mp4'/> </video>"
-        animated = "<img src='prediction.gif' alt='pv_loop'>"  #  "prediction.gif" # style="width:48px;height:48px;"
+        video = video.replace("avi", "mp4")
+        #video = "<video autoplay loop muted> <source src=video_file type='video/mp4'/> </video>"
+        animated =   "prediction.mp4" # style="width:48px;height:48px;" # "<img src='prediction.gif' alt='pv_loop'>"
         return video, animated, Rm, Ra, Emax, Emin, Vd, Tc, start_v
 
 
@@ -853,8 +853,8 @@ with gr.Blocks() as demo:
             
             generate_button = gr.Button("Load sample echocardiogram and generate result")
             with gr.Row():
-                video = gr.HTML() # PlayableVideo(autoplay=True) #format="avi"
-                plot =  gr.HTML() # gr.PlayableVideo(autoplay=True)
+                video =  gr.PlayableVideo(autoplay=True) #format="avi" gr.HTML()
+                plot =  gr.PlayableVideo(autoplay=True) # gr.HTML() # 
             
             with gr.Row():
                 Rm = gr.Number(label="Mitral valve circuit resistance (Rm) mmHg*s/ml:")

app.py

@@ -486,7 +486,7 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animate=True):
         
     if animate:
         anim = animation.FuncAnimation(fig, partial(update), frames=100, interval=30)
-        anim.save("prediction.gif")
+        anim.save("prediction.mp4")
         
     return plt, Rm, Ra, Emax, Emin, Vd, Tc, start_v
 
@@ -770,9 +770,9 @@ def generate_example():
         video = f"EchoNet-Dynamic/Videos/{filename}"
 
         plot, Rm, Ra, Emax, Emin, Vd,Tc, start_v = pvloop_simulator(Rm=round(results[4].item(),2), Ra=round(results[5].item(),2), Emax=round(results[2].item(),2), Emin=round(results[3].item(),2), Vd=round(results[6].item(),2), Tc=round(results[0].item(),2), start_v=round(results[1].item(),2))
-        video_file = video.replace("avi", "mp4")
-        video = "<video autoplay loop muted> <source src=video_file type='video/mp4'/> </video>"
-        animated = "<img src='prediction.gif' alt='pv_loop'>"  #  "prediction.gif" # style="width:48px;height:48px;"
+        video = video.replace("avi", "mp4")
+        #video = "<video autoplay loop muted> <source src=video_file type='video/mp4'/> </video>"
+        animated =   "prediction.mp4" # style="width:48px;height:48px;" # "<img src='prediction.gif' alt='pv_loop'>"
         return video, animated, Rm, Ra, Emax, Emin, Vd, Tc, start_v
 
 
@@ -853,8 +853,8 @@ with gr.Blocks() as demo:
             
             generate_button = gr.Button("Load sample echocardiogram and generate result")
             with gr.Row():
-                video = gr.HTML() # PlayableVideo(autoplay=True) #format="avi"
-                plot =  gr.HTML() # gr.PlayableVideo(autoplay=True)
+                video =  gr.PlayableVideo(autoplay=True) #format="avi" gr.HTML()
+                plot =  gr.PlayableVideo(autoplay=True) # gr.HTML() # 
             
             with gr.Row():
                 Rm = gr.Number(label="Mitral valve circuit resistance (Rm) mmHg*s/ml:")