Update model.py

model.py

@@ -12,6 +12,7 @@ from uuid import uuid4
 from PIL import Image
 
 from creds import get_credentials
+from io import BytesIO
 
 
 def generate_download_signed_url_v4(blob_name):
@@ -37,8 +38,8 @@ def generate_download_signed_url_v4(blob_name):
 
     # print("Generated GET signed URL:")
     # print(url)
-    # print("You can use this URL with any user agent, for example:")
-    # print(f"curl '{url}'")
+    print("You can use this URL with any user agent, for example:")
+    print(f"curl '{url}'")
     return url
 
 
@@ -59,8 +60,9 @@ class Model(LabelStudioMLBase):
 
 
             url = task["data"]["image"]
-            r = requests.get(generate_download_signed_url_v4(url), stream=True).raw
-            image = Image.open(r)
+            response = requests.get(generate_download_signed_url_v4(url))
+            image_data = BytesIO(response.read())
+            image = Image.open(image_data)
             
             original_width, original_height = image.size
             with torch.no_grad():