Anthony-Ml commited on
Commit
b953a35
1 Parent(s): ed2b731

Update app.py

Browse files

Added the last conv layer

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -92,7 +92,7 @@ with gr.Blocks() as demo:
92
  label = gr.Label(label="Predicted Class")
93
  with gr.Column():
94
  interpretation = gr.components.Interpretation(input_img)
95
- interpret.click(interpretation_function(input_img,learn, learn.layer4[-1],target_category=None), input_img, interpretation)
96
 
97
  #interpretation="default"
98
  enable_queue=True
 
92
  label = gr.Label(label="Predicted Class")
93
  with gr.Column():
94
  interpretation = gr.components.Interpretation(input_img)
95
+ interpret.click(interpretation_function(input_img,learn, learn.model[0][-1],target_category=None), input_img, interpretation)
96
 
97
  #interpretation="default"
98
  enable_queue=True