Update app.py

app.py

@@ -1,32 +1,32 @@
 import flask
-from flask import request, jsonify
+from flask import request
 import os
 from dotenv import load_dotenv
 load_dotenv()
+
 app = flask.Flask(__name__, template_folder="./")
+
+from transformers import pipeline
+
+classifier = pipeline('text-classification', model="bsenst/classify_services_model")
+
 @app.route('/')
 def index():
     return flask.render_template('index.html')
+
 @app.route("/", methods=["POST"])
 def predict():
     incoming = request.get_json()
     print(incoming)
-    # Ваша логика обработки текста здесь
-    return "Your response here"
+    prediction = classifier(incoming["text"])[0]
+    print(prediction)
+    return prediction
 @app.route("/avp", methods=["POST"])
 def avp():
     incoming = request.get_json()
     print(incoming)
     # Обработка POST запроса по пути "/avp"
     # Можно добавить нужную логику для обработки данных
-    
-    if "bazis" in incoming:
-        bazis = incoming["bazis"]
-        for user_id, score in bazis.items():
-            if score > 0:
-                bazis[user_id] = score - 1
-        incoming["bazis"] = bazis
-    
-    return jsonify(incoming)
+    return "test ok 200"
 if __name__ == '__main__':
     app.run(host='0.0.0.0',  port=int(os.environ.get('PORT', 7860)))