Update app.py

app.py

@@ -12,7 +12,7 @@ def get_audio_duration(audio_file):
     except wave.Error:
         raise ValueError("Invalid audio file. Please upload a valid .wav file.")
 
-def get_training_info(audio_file):
+def get_training_info(audio_file, pretrain):
     """Determine training parameters based on the duration of the audio file."""
     try:
         if not audio_file:
@@ -24,13 +24,13 @@ def get_training_info(audio_file):
             sample_rate = audio.getframerate()
 
         training_info = {
-            (0, 2): (150, 'TITAN'),
-            (2, 3): (200, 'TITAN'),
-            (3, 5): (250, 'TITAN'),
-            (5, 10): (300, 'normal'),
-            (10, 25): (500, 'Normal'),
-            (25, 45): (700, 'Normal'),
-            (45, 60): (1000, 'Normal')
+            (0, 2): (150, f'{pretrain}'),
+            (2, 3): (200, f'{pretrain}'),
+            (3, 5): (250, f'{pretrain}'),
+            (5, 10): (300, f'{pretrain}'),
+            (10, 25): (500, f'{pretrain}'),
+            (25, 45): (700, f'{pretrain}'),
+            (45, 60): (1000, f'{pretrain}')
         }
 
         for (min_duration, max_duration), (epochs, pretrain) in training_info.items():
@@ -44,9 +44,10 @@ def get_training_info(audio_file):
 with gr.Blocks(theme=gr.themes.Base(primary_hue="sky", secondary_hue="blue"), title="RVC TRAINING HELPER") as demo:
     with gr.Tab("Main Settings"):
         audio_input = gr.Audio(type="filepath", label="Your Audio here")
+        pretraina = gr.Dropdown(choices=["OV2","Normal","TITAN","Snowie V3","RIN E3","SnowieV3 X RIN_E3","ITALIA"])
         start_button = gr.Button("Start!")
         output_text = gr.Textbox(scale=3, label="Your Output here")
-        start_button.click(get_training_info, inputs=[audio_input], outputs=[output_text])
+        start_button.click(get_training_info, inputs=[audio_input,pretraina], outputs=[output_text])
     
     with gr.Tab("Credits"):
         gr.Markdown("### This code originally by [TheStinger](https://huggingface.co/TheStinger)<br>Edited by Blane187")