sponsorblock-ml / src /predict.py

Commit History

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