Update model

app.py

@@ -63,7 +63,8 @@ import tempfile
 from ultralytics import YOLO
 from moviepy import ImageSequenceClip
 from scipy.io import wavfile
-from avse_code import run_avse
+#from avse_code import run_avse
+from avse_code_hid64h16 import run_avse
 
 
 
@@ -85,7 +86,8 @@ CHUNK_SIZE_VIDEO = 75     # 25fps × 3 sec
 @spaces.GPU
 def run_avse_inference(video_path, audio_path):
     avse_model = AVSEModule()
-    avse_state_dict = torch.load("ckpts/ep220_0908.oat.ckpt")
+    #avse_state_dict = torch.load("ckpts/ep220_0908.oat.ckpt")
+    avse_state_dict = torch.load("ckpts/hid64h16_ep224.ckpt")
     avse_model.load_state_dict(avse_state_dict, strict=True)
     avse_model.to("cuda")
     avse_model.eval()