Update main.py

main.py

@@ -10,7 +10,7 @@ import cv2
 import numpy as np
 import base64
 
-from detectron_infer import predict as detectron_predict
+from yolo_Numbering import predict as predict_yolo
 from yolo_infer import predict_yolo
 
 app = FastAPI()