voxmenthe commited on
Commit
01ee265
·
verified ·
1 Parent(s): 3aaca16

Upload fine-tuned model, tokenizer, and supporting files for modernbert-imdb-sentiment

Browse files
Files changed (2) hide show
  1. inference.py +2 -1
  2. pytorch_model.bin +3 -0
inference.py CHANGED
@@ -96,7 +96,8 @@ class SentimentInference:
96
  self.model = AutoModelForSequenceClassification.from_pretrained(
97
  model_hf_repo_id,
98
  config=loaded_config,
99
- trust_remote_code=True
 
100
  )
101
  print(f"Model {model_hf_repo_id} loaded successfully from Hugging Face Hub.")
102
 
 
96
  self.model = AutoModelForSequenceClassification.from_pretrained(
97
  model_hf_repo_id,
98
  config=loaded_config,
99
+ trust_remote_code=True,
100
+ force_download=True # <--- TEMPORARY - remove when everything is working
101
  )
102
  print(f"Model {model_hf_repo_id} loaded successfully from Hugging Face Hub.")
103
 
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:106c846a077b9a1c445b0fa4b5d490d5e58a81270399da15064a4ee3a3b7e1ec
3
+ size 600856675