Update model.py

model.py

@@ -3,7 +3,6 @@ import logging
 import torch
 import datetime
 import requests
-from io import BytesIO
 
 from google.cloud import storage
 from transformers import AutoImageProcessor, AutoModelForObjectDetection
@@ -15,10 +14,6 @@ from PIL import Image
 from creds import get_credentials
 
 
-os.environ["GOOGLE_APPLICATION_CREDENTIALS"]= get_credentials()
-print("cred set")
-
-
 def generate_download_signed_url_v4(blob_name):
     """Generates a v4 signed URL for downloading a blob.
 
@@ -42,8 +37,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
 
 
@@ -61,8 +56,8 @@ class Model(LabelStudioMLBase):
 
 
             url = task["data"]["image"]
-            response = requests.get(generate_download_signed_url_v4(url))
-            image = Image.open(BytesIO(response.content))
+            image = Image.open(requests.get(url, stream=True).raw)
+            
             original_width, original_height = image.size
             with torch.no_grad():