From eda2acd1a4a026b6915065a7609cdd983559fb46 Mon Sep 17 00:00:00 2001
From: zcr <cgzhou@aidlab.hk>
Date: Fri, 27 Mar 2026 14:47:25 +0800
Subject: [PATCH] 1

---
 main.py | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/main.py b/main.py
index a6f4ae7..17331fb 100644
--- a/main.py
+++ b/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