Эх сурвалжийг харах

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

lxylxy123321 1 өдөр өмнө
parent
commit
d6d978de5b

+ 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 = []