Support model repos and bare paths in addition to buckets/ prefix

app.py

@@ -115,8 +115,17 @@ def sign(
     _check_admin(t, authorization)
     _check_bucket(bucket)
     path = path.lstrip("/")
-    full = f"buckets/{bucket}/{path}"
-    if not fs.exists(full):
+    # Try model repo first (no prefix), then datasets/, then buckets/
+    candidates = [f"{bucket}/{path}", f"datasets/{bucket}/{path}", f"buckets/{bucket}/{path}"]
+    full = None
+    for c in candidates:
+        try:
+            if fs.exists(c):
+                full = c
+                break
+        except Exception:
+            continue
+    if full is None:
         raise HTTPException(404, f"file not found: {bucket}/{path}")
     exp = int(time.time()) + ttl
     sig = _sign(bucket, path, exp)
@@ -146,10 +155,17 @@ def download(
     if not hmac.compare_digest(expected, sig):
         raise HTTPException(403, "bad signature")
 
-    full = f"buckets/{b}/{f}"
-    try:
-        info = fs.info(full)
-    except FileNotFoundError:
+    candidates = [f"{b}/{f}", f"datasets/{b}/{f}", f"buckets/{b}/{f}"]
+    full = None
+    info = None
+    for c in candidates:
+        try:
+            info = fs.info(c)
+            full = c
+            break
+        except Exception:
+            continue
+    if full is None:
         raise HTTPException(404, "file not found")
     size = info.get("size")