a

create_app.py

@@ -35,13 +35,15 @@ def load_models():
             raise  # Re-raise to see full traceback
         # QWEN_MODEL = AutoModelForCausalLM.from_pretrained(model_name).half()
         # QWEN_MODEL = QWEN_MODEL.to(device)
-        print("QWEN MODEL LOADED")
+        print(QWEN_MODEL)
+        print("QWEN MODEL LOADED", flush=True)
         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")
+        print(LONGFORMER_MODEL)
+        print("LONGFORMER MODEL LOADED", flush=True)
         LONGFORMER_MODEL.eval()
         MODELS_LOADED = True
         print("LOAD ENDED")