f8c0a542db6da7d19633d68c9a6f7ea150aa48a4,api/predict.py,,,#,6

Before Change

input_parser = MAX_API.parser()
input_parser.add_argument("image", type=FileStorage, location="files", required=True,
                          help="An image file (encoded as PNG or JPG/JPEG)")
input_parser.add_argument("threshold", type=float, default=0.7,
                          help="Probability threshold for including a detected object in the response in the range [0, 1] (default: 0.7). Lowering the threshold includes objects the model is less certain about.")


label_prediction = MAX_API.model("LabelPrediction", {
    "label_id": fields.String(required=False, description="Class label identifier"),

After Change

from werkzeug.datastructures import FileStorage
from core.model import ModelWrapper

model_label = MAX_API.model("ModelLabel", {
    "id": fields.String(required=True, description="Class label identifier"),
    "name": fields.String(required=True, description="Class label"),
})

labels_response = MAX_API.model("LabelsResponse", {
    "count": fields.Integer(required=True,

Instances