segmentation-backend

Sleeping

diegokauer commited on Dec 26, 2023

Commit

9d86353

1 Parent(s): c0d1e9d

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -33,7 +33,7 @@ class Model(LabelStudioMLBase):
             image_path = task["data"]["image"]
             image = Image(image_path)
-                original_width, original_height = image.size
             with torch.no_grad():
                 inputs = image_processor(images=image, return_tensors="pt")

             image_path = task["data"]["image"]
             image = Image(image_path)
+            original_width, original_height = image.size
             with torch.no_grad():
                 inputs = image_processor(images=image, return_tensors="pt")