debug
Browse files
app.py
CHANGED
|
@@ -26,6 +26,9 @@ METHOD_MAP = {
|
|
| 26 |
|
| 27 |
|
| 28 |
def generate_viz(image, method, class_index=None):
|
|
|
|
|
|
|
|
|
|
| 29 |
viz_method = METHOD_MAP[method]
|
| 30 |
viz = viz_method(TRANSFORM, image, class_index=class_index)
|
| 31 |
viz.savefig("visualization.png")
|
|
|
|
| 26 |
|
| 27 |
|
| 28 |
def generate_viz(image, method, class_index=None):
|
| 29 |
+
print(f"Image: {image.size}")
|
| 30 |
+
print(f"Method: {method}")
|
| 31 |
+
print(f"Class: {class_index}")
|
| 32 |
viz_method = METHOD_MAP[method]
|
| 33 |
viz = viz_method(TRANSFORM, image, class_index=class_index)
|
| 34 |
viz.savefig("visualization.png")
|