Trying #2

app.py

@@ -1,12 +1,10 @@
 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')
+model, tokenizer = model.get_model(), model.get_tokenizer()
 
 
 @app.route('/', methods=['GET', 'POST'])
@@ -15,10 +13,10 @@ def home():
         data = request.json
         user_input = data['text']
         # Use your model to classify the text
-        prediction = model.predict(loaded_model, user_input, tokenizer)
+        prediction = model.predict(model, user_input, tokenizer)
         return jsonify({'classification': prediction})
     return render_template('home.html')
 
 
 if __name__ == '__main__':
-    app.run()
+    app.run()