Блог

Я пытаюсь обучить модель nlp, а затем показать результат, используя python и Flask, в записной книжке Google Colab Python. На первом этапе я использую FinBert, финансовую модель, основанную на нейронной сети Bert. он получает text.txt файл, содержащий тексты новостей, а вывод — это настроение каждого текста. Вот проблема, когда я пытаюсь отобразить вывод на порту ‘0.0.0.0: 5000’, он не работает. Кстати, я запускаю свои коды в «Google colab».

 here is the code: 
from flask import Flask
from flask_cors import CORS
import sys
import optparse
import time
from flask import request
import sys
from finbert.finbert import predict
from pytorch_pretrained_bert.modeling import BertForSequenceClassification
import nltk

nltk.download('punkt')
app = Flask(__name__)
CORS(app)
start = int(round(time.time()))
model = BertForSequenceClassification.from_pretrained('./models/', num_labels=3, cache_dir=None)

@app.route("/",methods=['POST'])
def score():
    text=request.get_json()['text']
    return(predict(text, model).to_json(orient='records'))

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000, debug=False, threaded=True)
  

Я хочу проверить, связана ли проблема с моделью или нет, поэтому я удаляю часть flask и просто передаю некоторый текст в модель, и она работает нормально. Поэтому есть некоторая проблема с flask, которую я не могу оштрафовать (совершенно новая для flask).

Я не уверен, что это будет быстрым решением для вашего случая, но в прошлый раз, когда у меня возникла проблема, и я не смог найти больше решения, просто понижение версии flask решило проблему, для чего я потратил несколько дней, чтобы найти решение / ошибку

 !pip install flask==0.12.2 
  

Если это решение, заслуга должна принадлежать этому