swapniel99 commited on
Commit
875e0ac
·
1 Parent(s): 0118b47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ transform = transforms.Compose([
33
  inv_transform = transforms.Normalize(mean=[-2, -2, -2], std=[4, 4, 4])
34
 
35
 
36
- def image_classifier(input_image, top_classes=3, show_cam=True, target_layer=-1, transparency=0.5):
37
  input = transform(input_image).unsqueeze(0)
38
 
39
  output = model(input)
 
33
  inv_transform = transforms.Normalize(mean=[-2, -2, -2], std=[4, 4, 4])
34
 
35
 
36
+ def image_classifier(input_image, top_classes=3, show_cam=True, target_layer=-2, transparency=0.7):
37
  input = transform(input_image).unsqueeze(0)
38
 
39
  output = model(input)