Improve prediction pipeline
813b772
Joshua Lochner
commited on
Spaces:
Running
Running
Commit History
Ignore prediction if categorized as nothing by classifier and extractor
e77b67b
Joshua Lochner
commited on
Upgrade classifier to transformer-based model
36f7534
Joshua Lochner
commited on
Fix logging messages in predict script
4d4de75
Joshua Lochner
commited on
Initialize logging in each script
c4f250e
Joshua Lochner
commited on
Do not allow predictions to miss start of video
aa018be
Joshua Lochner
commited on
Fix conflicting `--no_cuda` argument
09cabec
Joshua Lochner
commited on
Use correct logger per script
e3d3d3f
Joshua Lochner
commited on
Add `no_cuda` argument to not use GPU
de9c8c4
Joshua Lochner
commited on
Remove redundant calls to change device
8981122
Joshua Lochner
commited on
Add `output_as_json` argument for inference
52340fc
Joshua Lochner
commited on
Abstract inference code
8b71088
Joshua Lochner
commited on
Improve caching and downloading of classifier for predictions
fb87012
Joshua Lochner
commited on
Raise ModelLoadError if model does not exist
dffef09
Joshua Lochner
commited on
Assign default model for predictions
8f0e2d8
Joshua Lochner
commited on
Fix import error
dbf7b4c
Joshua Lochner
commited on
Remove unused imports
bdfb4b1
Joshua Lochner
commited on
Move `seconds_to_time` to shared
bb58e90
Joshua Lochner
commited on
Use classifier category if transformer generates unknown category
ad7fc61
Joshua Lochner
commited on
Use multiclass classifier to filter predictions
25f1183
Joshua Lochner
commited on
Temporarily disable filtering of predictions using classifier
69fe24d
Joshua Lochner
commited on
Add functionality to predict self-promo and interaction reminders
90d1f68
Joshua Lochner
commited on
Improve prediction output format
74c1216
Joshua Lochner
commited on
Ensure `input_ids` are on the correct device when predicting
5dd37ab
Joshua Lochner
commited on
Fix prediction and evaluation arguments
9b9ffd0
Joshua Lochner
commited on
Add source code
5fbdd3c
Joshua Lochner
commited on