Upload app.py with huggingface_hub

app.py

@@ -88,9 +88,11 @@ async def set_collider(request: Request):
     prev = read_slot(sid) or {}
     write_slot(sid, {
         "style_tokens": tokens,
-        "temperature": float(body.get("temperature", 1.3)),
-        "top_k": int(body.get("top_k", 40)),
-        "cfg_musiccoca": float(body.get("cfg", 3.0)),
+        "temperature": float(body.get("temperature", 1.1)),
+        "top_k": int(body.get("top_k", 50)),
+        "cfg_musiccoca": float(body.get("cfg", 1.6)),
+        "cfg_notes": float(body.get("cfg_notes", 2.4)),
+        "cfg_drums": float(body.get("cfg_drums", 4.0)),
         "model": body.get("model", prev.get("model", "mrt2_base")),
         "buffer": int(body.get("buffer", 0)),
     })
@@ -137,7 +139,8 @@ def stream(session_id: str) -> str:
                 aset = set(active)
                 notes = [8 if i in aset else -1 for i in range(128)]
                 cfgs = [discretize_cfg(c.get("cfg_musiccoca", 3.0), 0.2, 40),
-                        discretize_cfg(c.get("cfg_notes", 2.4), 0.2, 40), discretize_cfg(1.0, 1.0, 8)]
+                        discretize_cfg(c.get("cfg_notes", 2.4), 0.2, 40),
+                        discretize_cfg(c.get("cfg_drums", 4.0), 1.0, 8)]
                 cond = model._conditioning((list(tokens) + [-1] * model.num_musiccoca)[:model.num_musiccoca],
                                            notes, drums, cfgs)
                 source = model.model.encode(cond).to(dt)