diff --git a/main.py b/main.py
index f4f3192..0dee236 100644
--- a/main.py
+++ b/main.py
@@ -15,13 +15,14 @@ from predict import predict_onnx,predict_onnx_pdl,predict_onnx_dfine
 from crop_image import crop_image_v3,save_path,save_fail_path,save_pass_path,validate_path
 
 PORT = 9645
+platform = os.name
 # --- 日志配置字典 ---
 LOGGING_CONFIG = {
     "version": 1,
     "disable_existing_loggers": False,
     "formatters": {
         "default": {
-            "()": "uvicorn._logging.DefaultFormatter",
+            "()": f"uvicorn.{'_logging' if platform== 'nt' else 'logging'}.DefaultFormatter",
             "fmt": "%(levelprefix)s %(asctime)s | %(message)s",
             "datefmt": "%Y-%m-%d %H:%M:%S",
         },
diff --git a/predict.py b/predict.py
index 42d1997..fcc191c 100644
--- a/predict.py
+++ b/predict.py
@@ -1,6 +1,5 @@
 import os
 import numpy as np
-from train import MyResNet18, data_transform
 from crop_image import crop_image, convert_png_to_jpg,draw_points_on_image,bytes_to_pil,validate_path
 import time
 from PIL import Image, ImageDraw
@@ -10,6 +9,7 @@ import onnxruntime as ort
 
 def predict(icon_image, bg_image):
     import torch
+    from train import MyResNet18, data_transform
     current_dir = os.path.dirname(os.path.abspath(__file__))
     model_path = os.path.join(current_dir, 'model', 'resnet18_38_0.021147585306924.pth')
     coordinates = [