back

app.py

@@ -4,8 +4,6 @@ import torchaudio
 from audiocraft.models import AudioGen
 from audiocraft.data.audio import audio_write
 import gradio as gr
-from gradio.routes import mount_gradio_app
-from fastapi import FastAPI
 
 OUTPUT_DIR = "audio_files"
 DEFAULT_DURATION = 5  # Default duration for audio generation in seconds
@@ -50,17 +48,4 @@ with gr.Blocks() as demo:
     generate_button.click(fn=generate_audio, inputs=[descriptions, duration_slider], outputs=output)
 
 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}
-
-app
+demo.launch(share=True)