Example lobe.ai tensor server using Flask

tesnorServer.py

import os
import logging

from PIL import Image
from flask import Flask, request, jsonify
from TensorModel import TensorModel

app = Flask(__name__)

logging.getLogger('werkzeug').setLevel(logging.WARNING)

modelTest = TensorModel(os.path.realpath(os.path.join(os.path.dirname(__file__), "subdir-path-to-model", "model")))

@app.route("/classify/test", methods=["POST"])
def classifyTest():
	image = Image.open(request.json["imageFilePath"])
	if image.mode != "RGB":
		image = image.convert("RGB")

	return jsonify(modelTest.predict(image))

app.run(host="127.0.0.1", port=8080, threaded=True)