fast api health

app.py

@@ -4,6 +4,7 @@ import torchaudio
 from audiocraft.models import AudioGen
 from audiocraft.data.audio import audio_write
 import gradio as gr
+from fastapi import FastAPI
 
 OUTPUT_DIR = "audio_files"
 DEFAULT_DURATION = 5  # Default duration for audio generation in seconds
@@ -47,5 +48,17 @@ with gr.Blocks() as demo:
 
     generate_button.click(fn=generate_audio, inputs=[descriptions, duration_slider], outputs=output)
 
-demo.queue(max_size=10, status_update_rate="auto")
-demo.launch(share=True)
+demo = demo.queue(max_size=10, status_update_rate="auto")
+
+app = FastAPI()
+
+# Mount Gradio app on "/"
+app = mount_gradio_app(app, demo, path="/")
+
+# Add healthcheck endpoint
+@app.get("/health")
+def health_check():
+    # Check if model is at least initialized (not required but useful)
+    status = "ready" if model is not None else "loading"
+    return {"status": status, "queue_active": True}
+