feat image2sketch triton部署

fix

app/api/api_image2sketch.py

@@ -1,12 +1,10 @@
-import json
 import logging
-import time
 
 from fastapi import APIRouter, HTTPException
 
 from app.schemas.image2sketch import Image2SketchModel
 from app.schemas.response_template import ResponseModel
-from app.service.image2sketch_2.server import processing_pipeline
+from app.service.lineart.service import LineArtService
 
 router = APIRouter()
 logger = logging.getLogger()
@@ -30,16 +28,9 @@ def image2sketch(request_item: Image2SketchModel):
         }
     """
     try:
-        start_time = time.time()
-        logger.info(f"image2sketch request item is : @@@@@@:{json.dumps(request_item.dict())}")
-        sketch_url = processing_pipeline(
-            image_url=request_item.image_url,
-            thickness=request_item.default_style,
-            sketch_bucket=request_item.sketch_bucket,
-            sketch_name=request_item.sketch_name
-        )
-        logger.info(f"run time is : {time.time() - start_time}")
+        service = LineArtService(request_item)
+        result_url = service.get_result()
     except Exception as e:
         logger.warning(f"image2sketch Run Exception @@@@@@:{e}")
         raise HTTPException(status_code=404, detail=str(e))
-    return ResponseModel(data=sketch_url)
+    return ResponseModel(data=result_url)