30 lines
604 B
Python
30 lines
604 B
Python
import logging
|
|
import os
|
|
|
|
import cv2
|
|
import uvicorn
|
|
from fastdeploy.serving.server import SimpleServer
|
|
import fastdeploy as fd
|
|
from .controllers import image
|
|
from .model import model_instance
|
|
|
|
logging.getLogger().setLevel(logging.INFO)
|
|
|
|
# Create server, setup REST API
|
|
app = SimpleServer()
|
|
app.register(
|
|
task_name="fd/yolov5s",
|
|
model_handler=fd.serving.handler.VisionModelHandler,
|
|
predictor=model_instance,
|
|
)
|
|
|
|
app.include_router(image.router)
|
|
|
|
|
|
def run_app():
|
|
uvicorn.run("image_identification.main:app", host="0.0.0.0", port=8000, reload=True)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_app()
|