Spaces:
Runtime error
Runtime error
Commit
·
f20afb2
1
Parent(s):
54afbf6
Upload folder using huggingface_hub
Browse files- src/myNLI.py +2 -2
src/myNLI.py
CHANGED
@@ -26,8 +26,8 @@ class FactChecker:
|
|
26 |
self.envir = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
|
27 |
|
28 |
# Load LLM
|
29 |
-
self.tokenizer = AutoTokenizer.from_pretrained("MoritzLaurer/mDeBERTa-v3-base-mnli-xnli") # LOAD mDEBERTa TOKENIZER
|
30 |
-
self.mDeBertaModel = AutoModel.from_pretrained(f"src/mDeBERTa (ft) V6/mDeBERTa-v3-base-mnli-xnli-{self.INPUT_TYPE}") # LOAD FINETUNED MODEL
|
31 |
# Load classifier model
|
32 |
self.checkpoints = torch.load(f"src/mDeBERTa (ft) V6/{self.INPUT_TYPE}.pt", map_location=self.envir)
|
33 |
|
|
|
26 |
self.envir = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
|
27 |
|
28 |
# Load LLM
|
29 |
+
self.tokenizer = AutoTokenizer.from_pretrained("MoritzLaurer/mDeBERTa-v3-base-mnli-xnli", use_auth_token=False) # LOAD mDEBERTa TOKENIZER
|
30 |
+
self.mDeBertaModel = AutoModel.from_pretrained(f"src/mDeBERTa (ft) V6/mDeBERTa-v3-base-mnli-xnli-{self.INPUT_TYPE}", use_auth_token=False) # LOAD FINETUNED MODEL
|
31 |
# Load classifier model
|
32 |
self.checkpoints = torch.load(f"src/mDeBERTa (ft) V6/{self.INPUT_TYPE}.pt", map_location=self.envir)
|
33 |
|