Create model.py

model.py

@@ -0,0 +1,12 @@
+from transformers import AutoModel, AutoTokenizer
+
+MODEL_NAME = "distilbert-base-uncased"
+
+def load_model():
+    tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+    model = AutoModel.from_pretrained(MODEL_NAME)
+    return tokenizer, model
+
+if __name__ == "__main__":
+    tokenizer, model = load_model()
+    print("Model loaded successfully")