getting spicy now

app.py

@@ -19,8 +19,8 @@ def rms_normalize(y, target_rms=0.05):
 
 def preprocess_audio(waveform):
     waveform_np = waveform.cpu().squeeze().numpy()  # Move to CPU before converting to NumPy
-    processed_waveform_np = rms_normalize(peak_normalize(waveform_np))
-    return torch.from_numpy(processed_waveform_np).unsqueeze(0).to(device)
+#   processed_waveform_np = rms_normalize(peak_normalize(waveform_np))
+    return torch.from_numpy(waveform_np).unsqueeze(0).to(device)
 
 def create_slices(song, sr, slice_duration, bpm, num_slices=5):
     song_length = song.shape[-1] / sr
@@ -113,7 +113,8 @@ def generate_music(seed, use_chords, chord_progression, prompt_duration, musicge
     slices = create_slices(song, sr, 35, bpm, num_slices=5)
 
     # Load the model
-    model_continue = MusicGen.get_pretrained(musicgen_model)
+    model_name = musicgen_model.split(" ")[0]
+    model_continue = MusicGen.get_pretrained(model_name)
 
     # Setting generation parameters
     model_continue.set_generation_params(
@@ -205,7 +206,15 @@ with gr.Blocks() as iface:
             use_chords = gr.Checkbox(label="Control Chord Progression", value=False)
             chord_progression = gr.Textbox(label="Chord Progression (e.g., Am CM Dm E7 Am)", visible=True)
             prompt_duration = gr.Dropdown(label="Prompt Duration (seconds)", choices=list(range(1, 11)), value=7)
-            musicgen_model = gr.Textbox(label="MusicGen Model", value="thepatch/vanya_ai_dnb_0.1")
+            musicgen_models = [
+                "thepatch/vanya_ai_dnb_0.1 (small)",
+                "thepatch/budots_remix (small)",
+                "thepatch/PhonkV2 (small)",
+                "thepatch/bleeps-medium (medium)",
+                "thepatch/hoenn_lofi (large)"
+            ]
+
+            musicgen_model = gr.Dropdown(label="MusicGen Model", choices=musicgen_models, value=musicgen_models[0])
             num_iterations = gr.Slider(label="Number of Iterations", minimum=1, maximum=10, step=1, value=3)
             bpm = gr.Slider(label="BPM", minimum=60, maximum=200, step=1, value=140)
             generate_button = gr.Button("Generate Music")