Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -141,7 +141,7 @@ def predict():
|
|
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]
|
|
|
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]
|