Update simpler_app.py

simpler_app.py

@@ -94,16 +94,34 @@ lyrics: {lyrics}
 """
     
     prompt = f"{instruction.strip()}</s>"    
-    outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
+    outputs = pipe(prompt, max_new_tokens=512, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
     pattern = r'\<\|system\|\>(.*?)\<\|assistant\|\>'
     cleaned_text = re.sub(pattern, '', outputs[0]["generated_text"], flags=re.DOTALL)
     
     print(f"SUGGESTED Lyrics: {cleaned_text}")
     return cleaned_text.lstrip("\n")
 
+from gradio_client import Client
+def generate_audio_ref(tags):
+
+    client = Client("declare-lab/mustango")
+    result = client.predict(
+		prompt=tags,
+		steps=200,
+		guidance=3,
+		api_name="/predict"
+    )
+    print(result)
+
+    return result
+
 def general_process(theme, tags, lyrics):
-    result = prepare_lyrics_with_llm(theme, tags, lyrics)
-    return None, result
+    gr.Info("Generating Lyrics")
+    lyrics_result = prepare_lyrics_with_llm(theme, tags, lyrics)
+    
+    gr.Info("Generating audio ref")
+    audio_ref = generate_audio_ref(tags)
+    return audio_ref, lyrics_result
 
     
 with gr.Blocks() as demo: