Update app.py

app.py

@@ -21,9 +21,9 @@ def generate_viz(image, class_index=None):
     print(f"Image: {image.size}")
     print(f"Class: {class_index}")
     
-    viz = do_explain(TRANSFORM, image, class_index=class_index)
+    viz，pred = do_explain(TRANSFORM, image, class_index=class_index)
     viz.savefig("visualization.png")
-    return Image.open("visualization.png").convert("RGB")
+    return Image.open("visualization.png").convert("RGB"), pred
 
 title = "Explain ViT 😊"
 
@@ -32,7 +32,8 @@ iface = gr.Interface(fn=generate_viz, inputs=[
        
         gr.Number(label="Class Index", info="Class index to explain"),
     ],
-    outputs=gr.Image(),
+    outputs=[gr.Image(),
+            gr.text(),]
     title=title,
     allow_flagging="never",
     cache_examples=True,