Update app.py

app.py

@@ -7,7 +7,7 @@ import scipy.io.wavfile
 import io
 import numpy as np
 
-# 1. Resource-Optimized Model Loading
+# 1. Model Loading
 MODEL_ID = "facebook/musicgen-small"
 device = "cpu"
 
@@ -23,8 +23,6 @@ def generate_core(prompt, duration):
     
     duration = min(int(duration), 30)
     inputs = processor(text=[prompt], padding=True, return_tensors="pt").to(device)
-    
-    # 50 tokens = 1 second
     max_tokens = int(duration * 50)
     
     with torch.no_grad():
@@ -34,8 +32,8 @@ def generate_core(prompt, duration):
     audio_data = audio_values[0, 0].cpu().numpy()
     return sampling_rate, audio_data
 
-# 2. FastAPI Engine
-app = FastAPI(title="MusicGen Automation API")
+# 2. FastAPI Setup
+app = FastAPI()
 
 @app.post("/generate")
 async def api_generate(prompt: str = Query(...), duration: int = Query(10)):
@@ -48,7 +46,7 @@ async def api_generate(prompt: str = Query(...), duration: int = Query(10)):
         raise HTTPException(status_code=500, detail=str(e))
 
 # 3. Gradio Interface
-with gr.Blocks(theme=gr.themes.Default()) as demo:
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🎵 MusicGen Automation Hub")
     with gr.Row():
         with gr.Column():
@@ -60,5 +58,7 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
     
     run_btn.click(generate_core, [p_in, d_in], a_out)
 
-# Mount Gradio and KILL the API schema generator that causes the 500 error
-app = gr.mount_gradio_app(app, demo, path="/", show_api=False)
+# 4. Mounting without the incompatible 'show_api' argument
+# We mount it at the root. The API documentation crash is bypassed by 
+# the explicit Query parameters in the FastAPI routes above.
+app = gr.mount_gradio_app(app, demo, path="/")