fix path
Browse files- my_app/__init__.py +1 -1
my_app/__init__.py
CHANGED
|
@@ -22,7 +22,7 @@ def load_models():
|
|
| 22 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 23 |
if not MODELS_LOADED:
|
| 24 |
LONGFORMER_TOKENIZER = LongformerTokenizer.from_pretrained('allenai/longformer-base-4096', device='auto')
|
| 25 |
-
config = LongformerConfig.from_json_file("
|
| 26 |
LONGFORMER_MODEL = CustomLongformerForSequenceClassification(config).from_pretrained('SFM2001/LongFormerScorer')
|
| 27 |
LONGFORMER_MODEL = LONGFORMER_MODEL.to(device)
|
| 28 |
LONGFORMER_MODEL.eval()
|
|
|
|
| 22 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 23 |
if not MODELS_LOADED:
|
| 24 |
LONGFORMER_TOKENIZER = LongformerTokenizer.from_pretrained('allenai/longformer-base-4096', device='auto')
|
| 25 |
+
config = LongformerConfig.from_json_file("Longformer_checkpoint/config.json")
|
| 26 |
LONGFORMER_MODEL = CustomLongformerForSequenceClassification(config).from_pretrained('SFM2001/LongFormerScorer')
|
| 27 |
LONGFORMER_MODEL = LONGFORMER_MODEL.to(device)
|
| 28 |
LONGFORMER_MODEL.eval()
|