Update app.py

app.py

@@ -14,14 +14,14 @@ print("Audiocraft version:", audiocraft.__version__)
 
 app = FastAPI()
 
-MODEL_NAME = "facebook/musicgen-medium"
+MODEL_NAME = "facebook/musicgen-small"
 OUTPUT_DIR = "outputs"
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
 print("Loading MusicGen model...")
 model = MusicGen.get_pretrained(MODEL_NAME)
 model.set_generation_params(
-    duration=30,
+    duration=10,
     temperature=1.0,
     top_k=250,
     top_p=0.0
@@ -42,21 +42,29 @@ class JobStatus:
     FAILED = "failed"
 
 
+import signal
+
+class GenerationTimeout(Exception):
+    pass
+
+def timeout_handler(signum, frame):
+    raise GenerationTimeout()
+
 def worker():
-    """Background worker that processes queued jobs"""
     while True:
         job_id = job_queue.get()
         job = jobs.get(job_id)
 
-        if not job:
-            job_queue.task_done()
-            continue
-
         try:
             jobs[job_id]["status"] = JobStatus.PROCESSING
 
+            signal.signal(signal.SIGALRM, timeout_handler)
+            signal.alarm(420)  # 7 minutes max
+
             wav = model.generate([job["prompt"]])[0]
 
+            signal.alarm(0)
+
             filename = f"{job_id}.wav"
             path = os.path.join(OUTPUT_DIR, filename)
 
@@ -73,13 +81,21 @@ def worker():
                 "file_path": path
             })
 
+        except GenerationTimeout:
+            jobs[job_id].update({
+                "status": JobStatus.FAILED,
+                "error": "Generation timed out (CPU limit)"
+            })
+
         except Exception as e:
             jobs[job_id].update({
                 "status": JobStatus.FAILED,
                 "error": str(e)
             })
 
-        job_queue.task_done()
+        finally:
+            job_queue.task_done()
+
 
 
 # Start worker thread