frist commit

router/image_clf.py

@@ -14,16 +14,6 @@ router = APIRouter()
 BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))  
 MODEL_PATH = os.path.join(BASE_DIR, "ml-models", "vit-human-pose-classification/")
 
-def download_image(url):
-    """Download and open an image from a URL."""
-    try:
-        response = requests.get(url)
-        response.raise_for_status()
-        return Image.open(BytesIO(response.content)).convert("RGB")
-    except Exception as e:
-        raise HTTPException(status_code=400, detail=f"Failed to download image: {e}")
-
-
 @router.post(
     "/image_classification",
     response_model=ImageOutput,
@@ -33,12 +23,12 @@ def download_image(url):
 def image_classification(input: ImageInput)-> ImageOutput:
     try:
         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}")
+        image = input.url
+        
+        logger.info(f"Image URLs: {image[-1]}")
         
         start = time.time()
-        output = pipe(urls)
+        output = pipe(image[-1])
         end = time.time()
         
         prediction_time = int((end-start)*1000)