1

main.py

@@ -1,17 +1,14 @@
-import io
 import os
-import urllib.request  # 必须这样写，不能只 import urllib
-import uuid
+import urllib.request
 from typing import Optional, List
 
 import cv2
 import litserve as ls
 import numpy as np
 import torch
-from PIL import Image, ImageDraw
+from PIL import Image
 from minio import Minio
 from pydantic import BaseModel, Field
-from fastapi import Response  # 导入 FastAPI 的 Response
 
 from config import settings
 from segment_anything import SamPredictor, sam_model_registry
@@ -21,7 +18,8 @@ minio_client = Minio(settings.MINIO_URL, access_key=settings.MINIO_ACCESS, secre
 
 
 class SAMRequest(BaseModel):
-    user_id: int
+    bucket: str = Field(...)
+    object_name: str = Field(...)
     image_path: str = Field(..., description="图片路径，必填字段")
     type: str = Field(..., description="推理类型，必填字段")
     points: Optional[List[List[float]]] | None = None