WwYc commited on
Commit
44c214e
·
verified ·
1 Parent(s): ead720f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def generate_viz(image, class_index=None):
21
  print(f"Image: {image.size}")
22
  print(f"Class: {class_index}")
23
 
24
- vizpred = do_explain(TRANSFORM, image, class_index=class_index)
25
  viz.savefig("visualization.png")
26
  return Image.open("visualization.png").convert("RGB"), pred
27
 
 
21
  print(f"Image: {image.size}")
22
  print(f"Class: {class_index}")
23
 
24
+ viz, pred = do_explain(TRANSFORM, image, class_index=class_index)
25
  viz.savefig("visualization.png")
26
  return Image.open("visualization.png").convert("RGB"), pred
27