Update utils.py

utils.py

@@ -1,7 +1,13 @@
-import tensorflow as tf
-from load_model import load_model
+import requests
+from dotenv import load_dotenv
+import os
 
-def predict(text):
-    model = load_model()
-    prediction = model.predict([text])
-    return prediction
+load_dotenv()
+api_key = os.getenv('HF_API_KEY')
+model_path = os.getenv('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()