Hugging Face's logo

Spaces:
diegokauer
/
segmentation-backend
Sleeping

App Files Community
diegokauer commited on
Commit
bced8ed
·
1 Parent(s): 8f40b79

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -23,7 +23,7 @@ class Model(LabelStudioMLBase):
23
 
24
 
25
  image_path = task["data"]["image"]
26
- image = Image(image_path)
27
  original_width, original_height = image.size
28
  with torch.no_grad():
29
 
 
23
 
24
 
25
  image_path = task["data"]["image"]
26
+ image = Image.open(image_path)
27
  original_width, original_height = image.size
28
  with torch.no_grad():
29