Spaces:
Sleeping
Sleeping
Nick Sorros
commited on
Commit
·
6f41674
1
Parent(s):
fd5a1b3
Fix tagged grants
Browse files- tag.py +1 -1
- tagged_grants.jsonl +0 -0
tag.py
CHANGED
|
@@ -30,7 +30,7 @@ def tag(data_path, tagged_data_path, sample_size: int = 10):
|
|
| 30 |
labels = model(**inputs, return_labels=True)
|
| 31 |
|
| 32 |
for i, tags in enumerate(labels):
|
| 33 |
-
data[i]["tags"] = tags
|
| 34 |
|
| 35 |
srsly.write_jsonl(tagged_data_path, data)
|
| 36 |
|
|
|
|
| 30 |
labels = model(**inputs, return_labels=True)
|
| 31 |
|
| 32 |
for i, tags in enumerate(labels):
|
| 33 |
+
data[batch_index+i]["tags"] = tags
|
| 34 |
|
| 35 |
srsly.write_jsonl(tagged_data_path, data)
|
| 36 |
|
tagged_grants.jsonl
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|