Update model_loading.py

model_loading.py

@@ -1,13 +1,9 @@
-import os
-import requests
-from dotenv import load_dotenv
+from load_model import load_model
 
-load_dotenv()
-api_key = os.getenv('HF_API_KEY')
-model_path = os.getenv('MODEL_PATH')
+model, tokenizer = load_model()
 
-def get_model_predictions(text):
-    headers = {"Authorization": f"Bearer {api_key}"}
-    payload = {"inputs": text}
-    response = requests.post(f"https://api.huggingface.co/models/{model_path}", headers=headers, json=payload)
-    return response.json()
+def get_model():
+    return model
+
+def get_tokenizer():
+    return tokenizer