app.py 文件源码

def main(_):
  model = Model()
  if FLAGS.train:
    model.train()
    model.save()
    sys.exit(0)
  else:
    model.restore()
    app = Flask(__name__)
    server = Server()
    server.set_model(model)
    app.add_url_rule('/', view_func=server.server_running)
    app.add_url_rule('/predict', view_func=server.predict)
    app.run(host= '0.0.0.0', port=80)
    sys.exit(0)