frist commit

router/image_clf.py

@@ -32,16 +32,18 @@ def image_classification(input: ImageInput)-> ImageOutput:
         end = time.time()
         
         prediction_time = int((end-start)*1000)
-        labels_and_scores = [{"label": x['label'], "score": x['score']} for x in output]
+        
+        labels = [x["label"] for x in output]
+        scores = [x["score"] for x in output]
 
         return ImageOutput(
             user_id=input.user_id,
             url=input.url,
             model_name="vit-human-pose-classification",
-            labels=[x['label'] for x in labels_and_scores],
-            scores=[x['score'] for x in labels_and_scores],
+            label=labels,
+            score=scores,
             prediction_time=prediction_time
         )
    
     except Exception as e:
-        return {"error": f"Failed to process image classification: {str(e)}"}, 500
+        raise HTTPException(status_code=500, detail=f"Failed to process image classification: {e}")