Update model.py

model.py

@@ -2,6 +2,8 @@ import os
 import logging
 import torch
 import datetime
+import requests
+from io import BytesIO
 
 from google.cloud import storage
 from transformers import AutoImageProcessor, AutoModelForObjectDetection
@@ -58,8 +60,9 @@ class Model(LabelStudioMLBase):
         for task in tasks:
 
 
-            image_path = task["data"]["image"]
-            image = Image.open(generate_download_signed_url_v4(image_path))
+            url = task["data"]["image"]
+            response = requests.get(generate_download_signed_url_v4(url))
+            image = Image.open(BytesIO(response.content))
             original_width, original_height = image.size
             with torch.no_grad():