Update app.py
Browse files
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/
|
| 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'])
|