from ultralytics import YOLO | |
from flask import Flask, request, jsonify | |
app = Flask(__name__) | |
model = YOLO('best.pt') | |
def predict(): | |
image = request.files['file'].read() | |
results = model(image) | |
return jsonify(results.pandas().xyxy[0].to_json(orient="records")) | |
if __name__ == '__main__': | |
app.run(host='0.0.0.0', port=8080) | |