Spaces:
Build error
Build error
Process images as numpy arrays
Browse files- vendiscore.py +4 -1
vendiscore.py
CHANGED
@@ -75,7 +75,10 @@ def get_features(config_name):
|
|
75 |
if config_name in ("text", "default"):
|
76 |
return datasets.Features({"samples": datasets.Value("string")})
|
77 |
if config_name == "image":
|
78 |
-
return
|
|
|
|
|
|
|
79 |
if config_name in ("K", "X"):
|
80 |
return [
|
81 |
datasets.Features(
|
|
|
75 |
if config_name in ("text", "default"):
|
76 |
return datasets.Features({"samples": datasets.Value("string")})
|
77 |
if config_name == "image":
|
78 |
+
return [
|
79 |
+
datasets.Features({"samples": datasets.Array2D}),
|
80 |
+
datasets.Features({"samples": datasets.Array3D}),
|
81 |
+
]
|
82 |
if config_name in ("K", "X"):
|
83 |
return [
|
84 |
datasets.Features(
|