lainlives commited on
Commit
2bdd960
·
1 Parent(s): b9bb559
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -18,6 +18,11 @@ from ultimate_rvc.rvc.infer.infer import VoiceConverter
18
  from ultimate_rvc.typing_extra import Json, StrPath
19
  import gradio as gr
20
  import lazy_loader as lazy
 
 
 
 
 
21
  import typer
22
  from huggingface_hub import snapshot_download
23
  from pydantic import BaseModel, ValidationError
@@ -3051,7 +3056,6 @@ def run_pipeline2(
3051
  noise_track, clean_track = separate_audio(
3052
  vocals_dereverb_track,
3053
  song_dir,
3054
- SeparationModel.UVR_DeNoise,
3055
  SegmentSize.SEG_2048,
3056
  )
3057
  display_progress("[~] Pitch-shifting vocals...", 5 / 9, progress_bar)
@@ -3572,7 +3576,7 @@ def render_song_cover_one_click_tab(
3572
  generate_btn.click(
3573
  partial(
3574
  exception_harness(
3575
- run_pipeline2,
3576
  info_msg="Song cover generated successfully!",
3577
  ),
3578
  cookiefile=cookiefile,
 
18
  from ultimate_rvc.typing_extra import Json, StrPath
19
  import gradio as gr
20
  import lazy_loader as lazy
21
+ from ultimate_rvc.core.generate.song_cover import (
22
+ get_named_song_dirs,
23
+ get_song_cover_name,
24
+ run_pipeline,
25
+ )
26
  import typer
27
  from huggingface_hub import snapshot_download
28
  from pydantic import BaseModel, ValidationError
 
3056
  noise_track, clean_track = separate_audio(
3057
  vocals_dereverb_track,
3058
  song_dir,
 
3059
  SegmentSize.SEG_2048,
3060
  )
3061
  display_progress("[~] Pitch-shifting vocals...", 5 / 9, progress_bar)
 
3576
  generate_btn.click(
3577
  partial(
3578
  exception_harness(
3579
+ run_pipeline,
3580
  info_msg="Song cover generated successfully!",
3581
  ),
3582
  cookiefile=cookiefile,