sna89 commited on
Commit
aba5ffb
·
1 Parent(s): 252d082

update app.py file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def segment_image(img):
24
  outputs = model(**img)
25
 
26
  predicted_semantic_map = processor.post_process_semantic_segmentation(
27
- outputs, target_sizes=[image.size[::-1]]
28
  )[0]
29
 
30
  fig, ax = plt.subplots(figsize=(5, 5))
 
24
  outputs = model(**img)
25
 
26
  predicted_semantic_map = processor.post_process_semantic_segmentation(
27
+ outputs, target_sizes=[img.size[::-1]]
28
  )[0]
29
 
30
  fig, ax = plt.subplots(figsize=(5, 5))