Update app.py

app.py

@@ -114,6 +114,7 @@ with gr.Blocks(css="#gradio-app-title { text-align: center; }") as demo:
                 ],
                 inputs=[selected_image, text_query],
                 outputs=[le_grad_output],
+                fn=viz_func,
                 label=''
             )