Update app.py

app.py

@@ -336,10 +336,8 @@ def test_fn(model, device, iteration, candidate_json_path, test_path, cfg, audio
         audio_low = model_out_candidates["audio_low"]
         # audio_high = model_out_candidates["audio_high_weight"]
         audio_high = model_out_candidates["audio_cls"].unsqueeze(0).repeat(1, audio_low.shape[1], 1)
-            
-        gap = target_length - np.concatenate(audio_low_list, axis=0).shape[1]
-        audio_low = F.normalize(audio_low, dim=2)[0][:gap].cpu().numpy()
-        audio_high = F.normalize(audio_high, dim=2)[0][:gap].cpu().numpy()
+        audio_low = F.normalize(audio_low, dim=2)[0].cpu().numpy()
+        audio_high = F.normalize(audio_high, dim=2)[0].cpu().numpy()
         audio_low_list.append(audio_low)
         audio_high_list.append(audio_high)
     
@@ -671,7 +669,7 @@ def make_demo():
               <br>
               News:
               <br>
-              [10/15]: Add watermark, fix bugs on custom character by downgrades to py3.9
+              [10/15]: Add watermark, fix bugs on custom character by downgrades to py3.9, fix bugs to support audio less than 4s.
               <br>
               [10/14]: Hugging face supports free L40S GPU for this project now!
               </div>