Exgc commited on
Commit
b8dba48
·
verified ·
1 Parent(s): 9a6ee98

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -119,7 +119,7 @@ def recover_waveform(mag_mix, phase_mix, pred_mask, args):
119
 
120
  def run_inference(input_audio, text_pos, audio_pos, image_pos, text_neg, audio_neg, image_neg,
121
  text_w, image_w, audio_w, neg_w):
122
- model, imagebind_net, args = setup_models("./exp/checkpoints/best_model.pt", "./exp/checkpoints/train-args.json")
123
  audio_len = 65535
124
  mag_mix, phase_mix, out_len = load_audio_and_spec(input_audio, audio_len,
125
  args['audio_rate'], args['n_fft'], args['hop_len'], args['win_len'])
 
119
 
120
  def run_inference(input_audio, text_pos, audio_pos, image_pos, text_neg, audio_neg, image_neg,
121
  text_w, image_w, audio_w, neg_w):
122
+ model, imagebind_net, args = setup_models("./exp/checkpoints/best_model.pt", "./exp/train-args.json")
123
  audio_len = 65535
124
  mag_mix, phase_mix, out_len = load_audio_and_spec(input_audio, audio_len,
125
  args['audio_rate'], args['n_fft'], args['hop_len'], args['win_len'])