DmitrMakeev commited on
Commit
f90e4c6
·
1 Parent(s): 0fa16fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -7
app.py CHANGED
@@ -1,19 +1,14 @@
1
  import flask
2
- from flask import request
3
  import os
4
  from dotenv import load_dotenv
5
  load_dotenv()
6
-
7
  app = flask.Flask(__name__, template_folder="./")
8
-
9
  from transformers import pipeline
10
-
11
  classifier = pipeline('text-classification', model="bsenst/classify_services_model")
12
-
13
  @app.route('/')
14
  def index():
15
  return flask.render_template('index.html')
16
-
17
  @app.route("/", methods=["POST"])
18
  def predict():
19
  incoming = request.get_json()
@@ -27,6 +22,14 @@ def avp():
27
  print(incoming)
28
  # Обработка POST запроса по пути "/avp"
29
  # Можно добавить нужную логику для обработки данных
30
- return "test ok 200"
 
 
 
 
 
 
 
 
31
  if __name__ == '__main__':
32
  app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))
 
1
  import flask
2
+ from flask import request, jsonify
3
  import os
4
  from dotenv import load_dotenv
5
  load_dotenv()
 
6
  app = flask.Flask(__name__, template_folder="./")
 
7
  from transformers import pipeline
 
8
  classifier = pipeline('text-classification', model="bsenst/classify_services_model")
 
9
  @app.route('/')
10
  def index():
11
  return flask.render_template('index.html')
 
12
  @app.route("/", methods=["POST"])
13
  def predict():
14
  incoming = request.get_json()
 
22
  print(incoming)
23
  # Обработка POST запроса по пути "/avp"
24
  # Можно добавить нужную логику для обработки данных
25
+
26
+ # Перебираем словарь с пользовательскими баллами
27
+ for user_id, score in incoming.items():
28
+ # Если балл больше 0, уменьшаем его на 1
29
+ if score > 0:
30
+ incoming[user_id] = score - 1
31
+
32
+ # Возвращаем измененный словарь
33
+ return jsonify(incoming)
34
  if __name__ == '__main__':
35
  app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))