Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 from dotenv import load_dotenv
-from flask import Flask
+from flask import Flask, request, jsonify
+import requests
 
 load_dotenv()
 api_key = os.getenv('HF_API_KEY')
@@ -8,9 +9,22 @@ model_path = os.getenv('MODEL_PATH')
 
 app = Flask(__name__)
 
-@app.route('/')
+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']
+    prediction = get_model_predictions(text)
+    return jsonify(prediction)
+
+@app.route('/', methods=['GET'])
 def index():
-    return "Welcome to My ai!"
+    return "Welcome to My AI!"
 
 if __name__ == '__main__':
-    app.run()
+    app.run(debug=True)