Lets hope final code update app.py

app.py

@@ -451,7 +451,29 @@ if st.button("Generate VLM Explanation"):
         if not st.session_state.get("vlm_response", False):
             try:
                 with st.spinner("Loading VLM Model. First time load will take time. Please be patient..."):
-                    
+                    try:
+                        vlm_info = load_vlm_model()
+                    except Exception as e:
+                        st.error("VLM load failed. See logs above.")
+                        vlm_info = None
+
+                if vlm_info is not None:
+                    try:
+                        img_for_vlm = overlay_pil.convert("RGB").resize((224, 224), Image.BILINEAR)
+                    except Exception:
+                        st.warning("Overlay image not available for VLM input; using original image.")
+                        img_for_vlm = pil_img.convert("RGB").resize((224, 224), Image.BILINEAR)
+
+                    with st.spinner("Generating Explanation...."):
+                        response = generate_vlm_response(
+                            vlm_info["processor"],
+                            vlm_info["model"],
+                            vlm_info["device"],
+                            img_for_vlm,
+                            pred_label,
+                            max_new_tokens=128
+                        )
+
                     response = "Debugging VLM response."
                     if response is None:
                         st.error("VLM did not return a response.")