Update load_model.py

load_model.py

@@ -1,9 +1,13 @@
 import os
-import tensorflow as tf
+import requests
 from dotenv import load_dotenv
 
 load_dotenv()
+api_key = os.getenv('HF_API_KEY')
 model_path = os.getenv('MODEL_PATH')
 
-def load_model():
-    return tf.keras.models.load_model(model_path)
+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()