updated

app.py

@@ -60,17 +60,19 @@ def main():
         image_class = predict_single_image(image, model, hp)
         #gradCam.save_and_display_gradcam()
         st.write(f"Image Class: {image_class}")
+
+        
         explanation = explainer.explain_instance(
             gray_img.astype('double'),
             model.predict,
-            top_labels=4,
+            top_labels=2,
             hide_color=0,
-            num_samples=500
+            num_samples=100
         )
         temp, mask = explanation.get_image_and_mask(
             explanation.top_labels[0],
             positive_only=True,
-            num_features=4,
+            num_features=5,
             hide_rest=True
         )
 
@@ -80,7 +82,7 @@ def main():
         # Save and display LIME explanation
         lime_explanation_path = 'lime_explanation.png'
         cv2.imwrite(lime_explanation_path, (xai * 255).astype(np.uint8))
-        st.image(lime_explanation_path, caption="LIME Explanation", use_column_width=True)
+        st.image((xai * 255).astype(np.uint8), caption="LIME Explanation", use_column_width=True)
         
 
 def convert_to_opencv(uploaded_file):