Spaces:
Running
Running
yacine
commited on
Commit
·
fbdf16b
1
Parent(s):
ed58320
filter_features for Sequence of ClassLabels
Browse files- tagging_app.py +8 -1
tagging_app.py
CHANGED
|
@@ -63,6 +63,12 @@ def filter_features(feature_dict):
|
|
| 63 |
"feature_type": feature_dict["_type"],
|
| 64 |
"feature": filter_features(feature_dict["feature"]),
|
| 65 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 66 |
else:
|
| 67 |
return dict(
|
| 68 |
[("feature_type", feature_dict["_type"])] + \
|
|
@@ -83,6 +89,7 @@ def filter_features(feature_dict):
|
|
| 83 |
else:
|
| 84 |
return dict([(k, filter_features(v)) for k, v in feature_dict.items()])
|
| 85 |
|
|
|
|
| 86 |
@st.cache
|
| 87 |
def find_languages(feature_dict):
|
| 88 |
if type(feature_dict) in [dict, datasets.features.Features]:
|
|
@@ -176,7 +183,7 @@ dataset_id = st.sidebar.selectbox(
|
|
| 176 |
|
| 177 |
if dataset_id == "local dataset":
|
| 178 |
path_to_info = st.sidebar.text_input("Please enter the path to the folder where the dataset_infos.json file was generated", "/path/to/dataset/")
|
| 179 |
-
if path_to_info
|
| 180 |
dataset_infos = json.load(open(pjoin(path_to_info, "dataset_infos.json")))
|
| 181 |
confs = dataset_infos.keys()
|
| 182 |
all_info_dicts = {}
|
|
|
|
| 63 |
"feature_type": feature_dict["_type"],
|
| 64 |
"feature": filter_features(feature_dict["feature"]),
|
| 65 |
}
|
| 66 |
+
elif "_type" in feature_dict["feature"] and feature_dict["feature"]["_type"] == "ClassLabel":
|
| 67 |
+
return {
|
| 68 |
+
"feature_type": feature_dict["_type"],
|
| 69 |
+
"dtype": "int32",
|
| 70 |
+
"feature": filter_features(feature_dict["feature"]),
|
| 71 |
+
}
|
| 72 |
else:
|
| 73 |
return dict(
|
| 74 |
[("feature_type", feature_dict["_type"])] + \
|
|
|
|
| 89 |
else:
|
| 90 |
return dict([(k, filter_features(v)) for k, v in feature_dict.items()])
|
| 91 |
|
| 92 |
+
|
| 93 |
@st.cache
|
| 94 |
def find_languages(feature_dict):
|
| 95 |
if type(feature_dict) in [dict, datasets.features.Features]:
|
|
|
|
| 183 |
|
| 184 |
if dataset_id == "local dataset":
|
| 185 |
path_to_info = st.sidebar.text_input("Please enter the path to the folder where the dataset_infos.json file was generated", "/path/to/dataset/")
|
| 186 |
+
if path_to_info not in ["/path/to/dataset/", ""]:
|
| 187 |
dataset_infos = json.load(open(pjoin(path_to_info, "dataset_infos.json")))
|
| 188 |
confs = dataset_infos.keys()
|
| 189 |
all_info_dicts = {}
|