vishal-keshav / InferenceServer

An ML inference server to serve Image classification over HTTP

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Develop the flask app api "class"

jessie-huo opened this issue · comments

User sends a request (http://127.0.0.1:5000/class?image=sample.jpg) and the code looks like this:

@app.route('/class', methods=['GET'])
def image():
# load densenet-121 model
# output = model(input)
# return output

Almost done. Needs to convert the output (a tensor) to string.
I'm not really sure if this is what we need. Please tell me what you think~