Spaces:

SuriC-nyc
/

MagicMusicMachine

Sleeping

suric commited on Mar 12, 2024

Commit

3e1d74a

1 Parent(s): 97a428f

fix bugs on prediction

Files changed (1) hide show

gradio_components/prediction.py CHANGED Viewed

@@ -92,6 +92,7 @@ def predict(model_path, text, melody, duration, topk, topp, temperature, target_
     model.set_custom_progress_callback(_progress)
     wavs = _do_predictions(
         [text],
         [melody],
         duration,

     model.set_custom_progress_callback(_progress)
     wavs = _do_predictions(
+        model,
         [text],
         [melody],
         duration,