route.py

@app.route("/classify")
@cross_origin()
def classify_symbol():
    json_str = request.args.get('points')
    if json_str is None:
        return {"top5": [" ", " ", " ", " ", " "]}

    json_data = json.loads(json_str)
    if 'data' not in json_data:
        return {"top5": [" ", " ", " ", " ", " "]}

    points = json_data['data']

    if len(points) == 1 and len(points[0]) == 0:
        return {"top5": [" ", " ", " ", " ", " "]}

    points_to_image(points)
    tensor = ToTensor()(Image.open("test.png").convert('RGB'))
    tensor = torch.unsqueeze(tensor, 0)
    output = model(tensor)
    pred = fix_predictions(output)

    return {"top5": pred}