dhruv2842 commited on
Commit
387a1a9
·
verified ·
1 Parent(s): d5a22d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -141,7 +141,7 @@ def predict():
141
  gradcam = GradCAM(model, target_layer_name)
142
 
143
  # Forward pass
144
- input_tensor.requires_grad = True
145
  output = model(input_tensor)
146
 
147
  probabilities = F.softmax(output, dim=1).cpu().numpy()[0]
 
141
  gradcam = GradCAM(model, target_layer_name)
142
 
143
  # Forward pass
144
+
145
  output = model(input_tensor)
146
 
147
  probabilities = F.softmax(output, dim=1).cpu().numpy()[0]