app.py

from flask import Flask, render_template, request, jsonify
import model  # Import your model module
from transformers import BertTokenizer

app = Flask(__name__)

# Load the model and tokenizer here
loaded_model = model.get_model()
tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')


@app.route('/', methods=['GET', 'POST'])
def home():
    if request.method == 'POST':
        data = request.json
        user_input = data['text']
        # Use your model to classify the text
        prediction = model.predict(loaded_model, user_input, tokenizer)
        return jsonify({'classification': prediction})
    return render_template('home.html')


if __name__ == '__main__':
    app.run()