Update load_model.py

load_model.py

@@ -1,12 +1,7 @@
-import os
-import requests
-from dotenv import load_dotenv
-
-load_dotenv()
-api_key = os.getenv('HF_API_KEY')
-model_path = os.getenv('MODEL_PATH')
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
 def load_model():
-    # For Hugging Face, loading a model is typically done via an API request rather than locally.
-    headers = {"Authorization": f"Bearer {api_key}"}
-    return headers, model_path
+    model_name = "Erfan11/Neuracraft"
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForCausalLM.from_pretrained(model_name)
+    return model, tokenizer