Hugging Face's logo

Spaces:
Conn-Finnegan
/
Soma_Skin_Cancer_Classifier
Sleeping

App Files Community
Conn Finnegan commited on
Commit
1e04ed9
·
verified ·
1 Parent(s): 944c22a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ transform = transforms.Compose([
21
  ])
22
 
23
  # Grad-CAM setup
24
- final_conv_layer = model.layer4[1].conv2 # Adjust if using a different architecture
25
  gradients = []
26
 
27
  def save_gradient(module, grad_input, grad_output):
 
21
  ])
22
 
23
  # Grad-CAM setup
24
+ final_conv_layer = model.layer3[1].conv2 # Adjust if using a different architecture
25
  gradients = []
26
 
27
  def save_gradient(module, grad_input, grad_output):