Update model_loader.py

model_loader.py

@@ -1,8 +1,5 @@
-from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
-
-def load_model():
-    model_id = "microsoft/phi-2"
-    tokenizer = AutoTokenizer.from_pretrained(model_id)
-    model = AutoModelForCausalLM.from_pretrained(model_id)
-    pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
-    return pipe
+from transformers import pipeline
+
+def load_model():
+    pipe = pipeline("text-generation", model="distilgpt2")
+    return pipe