فهرست منبع

修复图片数据集下载无记录问题

lxylxy123321 16 ساعت پیش
والد
کامیت
d6d978de5b
1فایلهای تغییر یافته به همراه6 افزوده شده و 0 حذف شده
  1. 6 0
      backend/app/services/dataset_service.py

+ 6 - 0
backend/app/services/dataset_service.py

@@ -324,6 +324,12 @@ def _download_modelscope_dataset(dataset_id: str) -> tuple[Path, Path, int]:
                     records.append(json.loads(line))
                 except json.JSONDecodeError:
                     continue
+    elif target.suffix == ".csv":
+        import csv as _csv
+        records = []
+        reader = _csv.DictReader(content.splitlines())
+        for row in reader:
+            records.append(dict(row))
     else:
         records = []