frist commit

router/image_clf.py

@@ -7,6 +7,7 @@ from io import BytesIO
 from fastapi import APIRouter, HTTPException
 from scripts.data_model import ImageInput, ImageOutput
 from utils.pipeline import load_model
+from utils.log import logger
 
 router = APIRouter()
 
@@ -34,6 +35,7 @@ def image_classification(input: ImageInput)-> ImageOutput:
         pipe = load_model(MODEL_PATH, is_image_model=True)
         # image = download_image(input.url)
         urls = [str(x) for x in input.url]
+        logger.info(f"Image URLs: {urls}")
         
         start = time.time()
         output = pipe(urls)
@@ -52,4 +54,4 @@ def image_classification(input: ImageInput)-> ImageOutput:
         )
    
     except Exception as e:
-        return {"error": f"Failed to process text classification: {str(e)}"}, 500
+        return {"error": f"Failed to process image classification: {str(e)}"}, 500