Front and back

app.py

@@ -12,8 +12,27 @@ model = None
 
 @spaces.GPU
 def generate_audio(descriptions: str):
-    print("generate audio called")
-    return None
+    global model
+    if model is None:
+        model = AudioGen.get_pretrained('facebook/audiogen-medium')
+        model.set_generation_params(duration=10) 
+
+    safe_name = "_".join(descriptions.split())
+    output_path = os.path.join(OUTPUT_DIR, safe_name)
+
+    wav = model.generate([descriptions]) 
+    audio_write(
+        output_path,
+        wav[0].cpu(),
+        model.sample_rate,
+        strategy="loudness",
+        loudness_compressor=True,
+        add_suffix=False,
+    )
+
+    final_path = f"{output_path}.wav"
+    print(f"Generated audio for '{descriptions}' -> {final_path}")
+    return final_path
 
 with gr.Blocks() as demo:
     gr.Markdown("# AudioGen Demo")