Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
viz
|
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 |
|