Update app.py

Add layer name

app.py

@@ -92,7 +92,7 @@ with gr.Blocks() as demo:
             label = gr.Label(label="Predicted Class")
         with gr.Column():
             interpretation = gr.components.Interpretation(input_img)
-    interpret.click(interpretation_function(input_img,learn, learn.model[-1],target_category=None), input_img, interpretation)
+    interpret.click(interpretation_function(input_img,learn, "_conv_head",target_category=None), input_img, interpretation)
 
 #interpretation="default"
 enable_queue=True