Upload fine-tuned model, tokenizer, and supporting files for modernbert-imdb-sentiment
Browse files- inference.py +2 -1
- 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
|