Update app.py

app.py

@@ -24,7 +24,7 @@ def generate_viz(image, class_index=None, use_threshold=False):
     image_trans = TRANSFORM(image)
     viz = generate_visualization(image_trans, class_index=class_index, use_threshold=use_threshold)
     
-    return Image.open("viz").convert("RGB")
+    return Image.open(viz).convert("RGB")
 
 title = "Explain ViT 😊"