small fixes

app.py

@@ -74,12 +74,11 @@ labels = [ "benign", "malignant", "normal" ]
 
 def predict(img):
     explainer = Explainer(model, img, labels)
-    #shap_img = explainer.shap()
-    return explainer.confidences
+    shap_img = explainer.shap()
+    return [explainer.confidences, shap_img]
 
 ui = gr.Interface(fn=predict, 
                 inputs=gr.Image(type="pil"),
-                outputs=gr.Label(num_top_classes=3),
-                interpretation="shap", num_shap=5
+                outputs=[gr.Label(num_top_classes=3), gr.Image(type="pil")],
                 examples=["benign (52).png", "benign (243).png", "malignant (127).png", "malignant (201).png", "normal (81).png", "normal (101).png"]).launch()
 ui.launch(share=True)