SFM2001 commited on
Commit
468932e
·
1 Parent(s): 89b9e00
Files changed (1) hide show
  1. 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("checkpoints/Longformer_checkpoint/config.json")
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()