imperialwool commited on
Commit
20c986e
·
1 Parent(s): 8357c1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -4
app.py CHANGED
@@ -11,7 +11,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification, Auto
11
 
12
  #initing
13
  app = Flask(__name__)
14
- VERSION = '1.0 build114'
15
  app.config['JSON_AS_ASCII'] = False
16
  limiter = Limiter(app=app, key_func=get_remote_address, default_limits=["5/minute"], storage_uri="memory://",)
17
 
@@ -101,7 +101,11 @@ chct_t, chct_m = AutoTokenizer.from_pretrained("cointegrated/rut5-small-chitchat
101
  @app.route('/analyzeText/api/v1/sentiment', methods=['GET', 'POST'])
102
  def sentimentAnalys():
103
  try:
104
- text = request.form.get('text') or request.args.get('text') or request.values.get('text') or request.json.get('text') or ""
 
 
 
 
105
  if text == "": return {"status": "error", "details": { "error_code": 101, "error_details": "No text provided" }}
106
 
107
  inputs = sa_t(text, return_tensors="pt")
@@ -117,7 +121,11 @@ def sentimentAnalys():
117
  @app.route('/analyzeText/api/v1/toxicity', methods=['GET', 'POST'])
118
  def toxicityAnalys():
119
  try:
120
- text = request.form.get('text') or request.args.get('text') or request.values.get('text') or request.json.get('text') or ""
 
 
 
 
121
  if text == "": return {"status": "error", "details": { "error_code": 101, "error_details": "No text provided" }} , 400
122
 
123
  inputs = tc_t(text, return_tensors="pt")
@@ -133,7 +141,11 @@ def toxicityAnalys():
133
  @app.route('/analyzeText/api/v1/chitchat', methods=['GET', 'POST'])
134
  def chitchatRu():
135
  try:
136
- text = request.form.get('text') or request.args.get('text') or request.values.get('text') or request.json.get('text') or ""
 
 
 
 
137
  if text == "": return {"status": "error", "details": { "error_code": 101, "error_details": "No text provided" }} , 400
138
 
139
  inputs = chct_t(text, padding=True, truncation=True, return_tensors="pt")
 
11
 
12
  #initing
13
  app = Flask(__name__)
14
+ VERSION = '1.0 build115'
15
  app.config['JSON_AS_ASCII'] = False
16
  limiter = Limiter(app=app, key_func=get_remote_address, default_limits=["5/minute"], storage_uri="memory://",)
17
 
 
101
  @app.route('/analyzeText/api/v1/sentiment', methods=['GET', 'POST'])
102
  def sentimentAnalys():
103
  try:
104
+ text = request.form.get('text') or request.args.get('text') or request.values.get('text') or ""
105
+ if text == "":
106
+ try: text = request.json.get('text') or ""
107
+ except: pass
108
+
109
  if text == "": return {"status": "error", "details": { "error_code": 101, "error_details": "No text provided" }}
110
 
111
  inputs = sa_t(text, return_tensors="pt")
 
121
  @app.route('/analyzeText/api/v1/toxicity', methods=['GET', 'POST'])
122
  def toxicityAnalys():
123
  try:
124
+ text = request.form.get('text') or request.args.get('text') or request.values.get('text') or ""
125
+ if text == "":
126
+ try: text = request.json.get('text') or ""
127
+ except: pass
128
+
129
  if text == "": return {"status": "error", "details": { "error_code": 101, "error_details": "No text provided" }} , 400
130
 
131
  inputs = tc_t(text, return_tensors="pt")
 
141
  @app.route('/analyzeText/api/v1/chitchat', methods=['GET', 'POST'])
142
  def chitchatRu():
143
  try:
144
+ text = request.form.get('text') or request.args.get('text') or request.values.get('text') or ""
145
+ if text == "":
146
+ try: text = request.json.get('text') or ""
147
+ except: pass
148
+
149
  if text == "": return {"status": "error", "details": { "error_code": 101, "error_details": "No text provided" }} , 400
150
 
151
  inputs = chct_t(text, padding=True, truncation=True, return_tensors="pt")