Blane187 commited on
Commit
2351560
·
verified ·
1 Parent(s): 4030926

Update src/webui.py

Browse files
Files changed (1) hide show
  1. src/webui.py +28 -32
src/webui.py CHANGED
@@ -221,8 +221,8 @@ if __name__ == '__main__':
221
  reverb_damping = gr.Slider(0, 1, value=0.7, label='Damping level', info='Absorption of high frequencies in the reverb')
222
 
223
  output_format = gr.Dropdown(['mp3', 'wav'], value='mp3', label='Output file type', info='mp3: small file size, decent quality. wav: Large file size, best quality')
224
-
225
- clear_btn = gr.ClearButton(value='Clear Settings', components=[song_input, rvc_model, keep_files, local_file])
226
 
227
  with gr.Row():
228
  generate_btn = gr.Button("Generate", variant='primary')
@@ -243,20 +243,17 @@ if __name__ == '__main__':
243
  protect, f0_method, crepe_hop_length, pitch_all, reverb_rm_size, reverb_wet,
244
  reverb_dry, reverb_damping, output_format, ai_cover])
245
 
246
- # Download tab
247
- with gr.Tab('Download model'):
248
-
249
- with gr.Tab('From HuggingFace/Pixeldrain URL'):
250
- with gr.Row():
251
- model_zip_link = gr.Text(label='Download link to model', info='Should be a zip file containing a .pth model file and an optional .index file.')
252
- model_name = gr.Text(label='Name your model', info='Give your new model a unique name from your other voice models.')
253
-
254
- download_btn = gr.Button('Download 🌐', variant='primary', scale=19)
255
- with gr.Row():
256
- dl_output_message = gr.Text(label='Output Message', interactive=False, scale=20)
257
 
258
- download_btn.click(download_online_model, inputs=[model_zip_link, model_name], outputs=dl_output_message)
259
-
260
  gr.Markdown('## Input Examples')
261
  gr.Examples(
262
  [
@@ -297,23 +294,22 @@ if __name__ == '__main__':
297
  download_pub_btn.click(download_online_model, inputs=[pub_zip_link, pub_model_name], outputs=pub_dl_output_message)
298
 
299
 
300
- with gr.Tab('Upload model'):
301
- gr.Markdown('## Upload locally trained RVC v2 model and index file')
302
- gr.Markdown('- Find model file (weights folder) and optional index file (logs/[name] folder)')
303
- gr.Markdown('- Compress files into zip file')
304
- gr.Markdown('- Upload zip file and give unique name for voice')
305
- gr.Markdown('- Click Upload model')
306
-
307
- with gr.Row():
308
- with gr.Column():
309
- zip_file = gr.File(label='Zip file')
310
-
311
- local_model_name = gr.Text(label='Model name')
312
-
313
- with gr.Row():
314
- model_upload_button = gr.Button('Upload model', variant='primary', scale=19)
315
- local_upload_output_message = gr.Text(label='Output Message', interactive=False, scale=20)
316
- model_upload_button.click(upload_local_model, inputs=[zip_file, local_model_name], outputs=local_upload_output_message)
317
 
318
  app.launch(
319
  share=args.share_enabled,
 
221
  reverb_damping = gr.Slider(0, 1, value=0.7, label='Damping level', info='Absorption of high frequencies in the reverb')
222
 
223
  output_format = gr.Dropdown(['mp3', 'wav'], value='mp3', label='Output file type', info='mp3: small file size, decent quality. wav: Large file size, best quality')
224
+ with gr.Column():
225
+ clear_btn = gr.ClearButton(value='Clear Settings', components=[song_input, rvc_model, keep_files, local_file])
226
 
227
  with gr.Row():
228
  generate_btn = gr.Button("Generate", variant='primary')
 
243
  protect, f0_method, crepe_hop_length, pitch_all, reverb_rm_size, reverb_wet,
244
  reverb_dry, reverb_damping, output_format, ai_cover])
245
 
246
+ # Download tab
247
+ with gr.Tab('Download model'):
248
+ with gr.Tab('From HuggingFace/Pixeldrain URL'):
249
+ with gr.Row():
250
+ model_zip_link = gr.Text(label='Download link to model', info='Should be a zip file containing a .pth model file and an optional .index file.')
251
+ model_name = gr.Text(label='Name your model', info='Give your new model a unique name from your other voice models.')
252
+ download_btn = gr.Button('Download 🌐', variant='primary', scale=19)
253
+ with gr.Row():
254
+ dl_output_message = gr.Text(label='Output Message', interactive=False, scale=20)
255
+ download_btn.click(download_online_model, inputs=[model_zip_link, model_name], outputs=dl_output_message)
 
256
 
 
 
257
  gr.Markdown('## Input Examples')
258
  gr.Examples(
259
  [
 
294
  download_pub_btn.click(download_online_model, inputs=[pub_zip_link, pub_model_name], outputs=pub_dl_output_message)
295
 
296
 
297
+ with gr.Tab('Upload model'):
298
+ gr.Markdown('## Upload locally trained RVC v2 model and index file')
299
+ gr.Markdown('- Find model file (weights folder) and optional index file (logs/[name] folder)')
300
+ gr.Markdown('- Compress files into zip file')
301
+ gr.Markdown('- Upload zip file and give unique name for voice')
302
+ gr.Markdown('- Click Upload model')
303
+
304
+ with gr.Row():
305
+ with gr.Column():
306
+ zip_file = gr.File(label='Zip file')
307
+
308
+ local_model_name = gr.Text(label='Model name')
309
+ with gr.Row():
310
+ model_upload_button = gr.Button('Upload model', variant='primary', scale=19)
311
+ local_upload_output_message = gr.Text(label='Output Message', interactive=False, scale=20)
312
+ model_upload_button.click(upload_local_model, inputs=[zip_file, local_model_name], outputs=local_upload_output_message)
 
313
 
314
  app.launch(
315
  share=args.share_enabled,