File size: 702 Bytes
634cec7 5cf5335 0ac9724 16ede2d 5cf5335 16ede2d 5cf5335 16ede2d 0ac9724 16ede2d 5cf5335 0ac9724 16ede2d 5cf5335 0ac9724 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import os
from flask import Flask, request, jsonify
from dotenv import load_dotenv
import requests
load_dotenv()
api_key = os.getenv('HF_API_KEY')
model_path = os.getenv('MODEL_PATH')
app = Flask(__name__)
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)
if __name__ == '__main__':
app.run(debug=True) |