Create load_model.py

load_model.py

@@ -0,0 +1,7 @@
+import torch
+from transformers import AutoModelForSequenceClassification
+
+def load_model(model_path):
+    model = AutoModelForSequenceClassification.from_pretrained(model_path)
+    model.eval()
+    return model