Update model.py

model.py

@@ -9,6 +9,38 @@ from uuid import uuid4
 from PIL import Image
 
 
+import datetime
+
+from google.cloud import storage
+
+def generate_download_signed_url_v4(blob_name):
+    """Generates a v4 signed URL for downloading a blob.
+
+    Note that this method requires a service account key file. You can not use
+    this if you are using Application Default Credentials from Google Compute
+    Engine or from the Google Cloud SDK.
+    """
+    bucket_name = os.getenv("bucket")
+
+    storage_client = storage.Client()
+    bucket = storage_client.bucket(bucket_name)
+    blob = bucket.blob(blob_name)
+
+    url = blob.generate_signed_url(
+        version="v4",
+        # This URL is valid for 15 minutes
+        expiration=datetime.timedelta(minutes=15),
+        # Allow GET requests using this URL.
+        method="GET",
+    )
+
+    print("Generated GET signed URL:")
+    print(url)
+    print("You can use this URL with any user agent, for example:")
+    print(f"curl '{url}'")
+    return url
+
+
 class Model(LabelStudioMLBase):
 
     image_processor = AutoImageProcessor.from_pretrained("diegokauer/conditional-detr-coe-int")
@@ -23,7 +55,7 @@ class Model(LabelStudioMLBase):
 
 
             image_path = task["data"]["image"]
-            image = Image.open(image_path)
+            image = Image.open(generate_download_signed_url_v4(image_path))
             original_width, original_height = image.size
             with torch.no_grad():