Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 from flask import Flask, request, jsonify
 from dotenv import load_dotenv
-import tensorflow as tf
+import requests
 
 load_dotenv()
 api_key = os.getenv('HF_API_KEY')
@@ -9,16 +9,18 @@ model_path = os.getenv('MODEL_PATH')
 
 app = Flask(__name__)
 
-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()
 
 @app.route('/predict', methods=['POST'])
 def predict():
     data = request.get_json()
     text = data['text']
-    model = load_model()
-    prediction = model.predict([text])
-    return jsonify(prediction.tolist())
+    prediction = get_model_predictions(text)
+    return jsonify(prediction)
 
 if __name__ == '__main__':
-    app.run()
+    app.run(debug=True)