frist commit

router/image_clf.py

@@ -32,10 +32,11 @@ 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)
+        # image = download_image(input.url)
+        urls = [str(x) for x in input.url]
         
         start = time.time()
-        output = pipe(image)
+        output = pipe(urls)
         end = time.time()
         
         prediction_time = int((end-start)*1000)