segmentation-backend

Sleeping

diegokauer commited on Dec 27, 2023

Commit

77a1ede

1 Parent(s): b4d11a3

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -59,7 +59,9 @@ class Model(LabelStudioMLBase):
             url = task["data"]["image"]
-            image = Image.open(requests.get(generate_download_signed_url_v4(url), stream=True).raw)
             original_width, original_height = image.size
             with torch.no_grad():

             url = task["data"]["image"]
+            r = requests.get(generate_download_signed_url_v4(url), stream=True).raw
+            print(r)
+            image = Image.open(r)
             original_width, original_height = image.size
             with torch.no_grad():