govtech
/

lionguard-v1

Model card Files Files and versions

shaunkhoo commited on Jul 16, 2024

Commit

d8fdbcd

·

1 Parent(s): 97f9a3c

fix: correct typo

Files changed (1) hide show

inference.py +1 -1

inference.py CHANGED Viewed

@@ -51,7 +51,7 @@ def predict(batch_text):
     for category, details in config['classifier'].items():
         # Download the classifier from HuggingFace hub
-        local_model_fp = hf_hub_download(repo_id = repo_path, filename = config['classifer'][category]['model_fp'])
         # Run the inference
         session = rt.InferenceSession(local_model_fp)

     for category, details in config['classifier'].items():
         # Download the classifier from HuggingFace hub
+        local_model_fp = hf_hub_download(repo_id = repo_path, filename = config['classifier'][category]['model_fp'])
         # Run the inference
         session = rt.InferenceSession(local_model_fp)