Upload 2 files

Main.py

@@ -33,7 +33,7 @@ main_path = data['MainPath']
 device = data['Device']
 diag_attn = data['DiagAttn']
 
-BestModelPath = 'Best_GlobalModel_500_0_0.pt'
+BestModelPath = '/home/siddc/SiddC/Misc/WatchMeSpeak/WatchMeSpeak/Best_GlobalModel_500_0_0.pt'
 
 def pre_process_mfcc(mfcc):
     

app.py

@@ -12,7 +12,7 @@ from cv2 import resize, INTER_LINEAR
 from PIL import Image
 
 import scipy.signal as signal
-from matplotlib.animation import FuncAnimation
+from matplotlib.animation import FuncAnimation, FFMpegWriter
 
 import torch
 import librosa
@@ -22,7 +22,7 @@ import streamlit as st
 from PIL import Image
 import streamlit as st
 
-im = Image.open("Logo.png")
+im = Image.open("/home/siddc/SiddC/Misc/WatchMeSpeak/WatchMeSpeak/Logo.png")
 st.set_page_config(
     page_title="Watch Me Talk",
     page_icon=im,
@@ -802,7 +802,11 @@ if what == 'Upload audio file':
         start = time.time()
         anim2 = FuncAnimation(fig, animate, frames = len(ema), interval = 10)
 
-        anim2.save('video.mp4', writer = 'ffmpeg', fps = 30)
+        # anim2.save('video.mp4', writer = 'ffmpeg', fps = 30)
+                
+        f = 'video.mp4'
+        writervideo = FFMpegWriter(fps=30) 
+        anim2.save(f, writer=writervideo)
 
         my_bar.progress(85)