Update app.py

app.py

@@ -10,7 +10,7 @@ def home():
         data = request.json
         user_input = data['text']
         # Use your model to classify the text
-        prediction = model.classify_text(user_input)
+        prediction = model.predict(user_input)
         return jsonify({'classification': prediction})
     return render_template('home.html')