KingNish commited on
Commit
d7227ce
·
verified ·
1 Parent(s): a75bb02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -447,7 +447,7 @@ def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=
447
  # Execute the command
448
  try:
449
  music = generate_music(genre_txt=genre_txt_content, lyrics_txt=lyrics_txt_content, run_n_segments=num_segments,
450
- output_dir=output_dir, cuda_idx=0, max_new_tokens=max_new_tokens)
451
  return music
452
  except Exception as e:
453
  gr.Warning("An Error Occured: " + str(e))
 
447
  # Execute the command
448
  try:
449
  music = generate_music(genre_txt=genre_txt_content, lyrics_txt=lyrics_txt_content, run_n_segments=num_segments,
450
+ output_dir=output_dir, max_new_tokens=max_new_tokens)
451
  return music
452
  except Exception as e:
453
  gr.Warning("An Error Occured: " + str(e))