Update app.py

app.py

@@ -27,7 +27,7 @@ from infer_tools.infer_tool import *
 import io
 
 clip_completed = False
-def render_audio(ckpt_temp_file, config_temp_file, audio_temp_file, title, title2, Crepe, noise_step, use_mel_as_base):
+def render_audio(ckpt_temp_file, config_temp_file, audio_temp_file, title, title2, choice, noise_step, use_mel_as_base):
     logging.getLogger('numba').setLevel(logging.WARNING)
     title = int(title)
     title2 = int(title2)
@@ -47,7 +47,7 @@ def render_audio(ckpt_temp_file, config_temp_file, audio_temp_file, title, title
     
     # Show the spinner and run the run_clip function inside the 'with' block
     with st.spinner("Rendering Audio..."):
-      f0_tst, f0_pred, audio = run_clip(svc_model,file_path=wav_fn, key=key, acc=title2, use_crepe=Crepe, use_pe=False, thre=0.05,
+      f0_tst, f0_pred, audio = run_clip(svc_model,file_path=wav_fn, key=key, acc=title2, use_crepe=choice, use_pe=False, thre=0.05,
                                         use_gt_mel=use_mel_as_base, add_noise_step=noise_step,project_name=project_name,out_path=wav_gen)
     clip_completed = True
     if clip_completed:
@@ -112,7 +112,7 @@ if "audio_temp_file" in locals():
 title = col2.number_input("Key", value=0, step=1, min_value=-12, max_value=12)
 title2 = col2.number_input("Speedup", value=20, step=1, min_value=5, max_value=100)
 title3 = col2.number_input("Gender Flag", value=1.00, step=0.01, min_value=0.70, max_value=1.30, help='Default is 1.0, it works by decimals, setting it at 1.05 will make your render sound more female-ish, setting it to 0.95 will make it sound more masculine, for example.')
-choice = col2.selectbox('Resampler', ('Crepe', 'Harvest'))
+choice = col2.checkbox('Use Crepe', value=False)
 # Create checkbox for using Mel as Base
 use_mel_as_base = col2.checkbox('Use Mel as Base', value=False, help='gt mel: Enabling this will use the input audio as a base and will unlock a new parameter, do not use this if you dont know what it does.')
 noise_step = 600
@@ -123,15 +123,10 @@ else:
     noise_step = 600
 password = col2.text_input("Enter password", help='Password can be got by agreeing to TOS and getting allowed after validation, you can go to the TOS here:')
 correct_password = os.environ.get("gatepassword")
-if choice == Crepe
-    Crepe = True
-else:
-    Crepe = False
-
 ###
 if st.button("Render audio"):
     if password == correct_password:
-        render_audio(ckpt_temp_file, config_temp_file, audio_temp_file, title, title2, noise_step, Crepe, use_mel_as_base)
+        render_audio(ckpt_temp_file, config_temp_file, audio_temp_file, title, title2, noise_step, choice, use_mel_as_base)
   
     else:
         st.error("Incorrect password")