a

create_app.py

@@ -38,11 +38,13 @@ def load_models():
         print("QWEN MODEL LOADED")
         LONGFORMER_TOKENIZER = LongformerTokenizer.from_pretrained('allenai/longformer-base-4096', device='auto')
         config = LongformerConfig.from_json_file("Longformer_checkpoint/config.json")
+        print("LONGFORMER TOKENIZER LOADED")
         LONGFORMER_MODEL = CustomLongformerForSequenceClassification(config).from_pretrained('SFM2001/LongFormerScorer')
         LONGFORMER_MODEL = LONGFORMER_MODEL.to(device)
+        print("LONGFORMER MODEL LOADED")
         LONGFORMER_MODEL.eval()
         MODELS_LOADED = True
-    print("LOAD ENDED")
+        print("LOAD ENDED")
 
 def create_app():
     set_seed(42)