|
import os |
|
from flask import Flask, request, jsonify |
|
from dotenv import load_dotenv |
|
import tensorflow as tf |
|
|
|
load_dotenv() |
|
api_key = os.getenv('HF_API_KEY') |
|
model_path = os.getenv('MODEL_PATH') |
|
|
|
app = Flask(__name__) |
|
|
|
def load_model(): |
|
return tf.keras.models.load_model(model_path) |
|
|
|
@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()) |
|
|
|
if __name__ == '__main__': |
|
app.run() |
|
|
