Views
No views yet
hazard_model.pkl – predicts the hazard category from title + text.product_model.pkl – predicts the product category (conditional on hazard prediction).title + text as a single string.1import joblib
2from huggingface_hub import hf_hub_download
3
4# Hazard model
5hazard_path = hf_hub_download(repo_id='Samrudhi013/st2-linear-svc', filename='hazard_model.pkl')
6hazard_model = joblib.load(hazard_path)
7
8# Product model
9product_path = hf_hub_download(repo_id='Samrudhi013/st2-linear-svc', filename='product_model.pkl')
10product_model = joblib.load(product_path)