Spaces:
Running
Running
Update clean.py
Browse files
clean.py
CHANGED
|
@@ -98,6 +98,9 @@ def main():
|
|
| 98 |
columns=["filename", "video_id", "video_path", "label", "start", "end"],
|
| 99 |
)
|
| 100 |
|
|
|
|
|
|
|
|
|
|
| 101 |
cleaned.to_csv(OUTPUT_CSV, index=False)
|
| 102 |
print(f"✅ Wrote {len(cleaned)} rows -> {OUTPUT_CSV}")
|
| 103 |
|
|
|
|
| 98 |
columns=["filename", "video_id", "video_path", "label", "start", "end"],
|
| 99 |
)
|
| 100 |
|
| 101 |
+
cleaned["start"] = cleaned["start"].astype("Int64")
|
| 102 |
+
cleaned["end"] = cleaned["end"].astype("Int64")
|
| 103 |
+
|
| 104 |
cleaned.to_csv(OUTPUT_CSV, index=False)
|
| 105 |
print(f"✅ Wrote {len(cleaned)} rows -> {OUTPUT_CSV}")
|
| 106 |
|