theschoolofai commited on
Commit
43a409b
·
1 Parent(s): fe0372f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ inv_normalize = transforms.Normalize(
18
  classes = ('plane', 'car', 'bird', 'cat', 'deer',
19
  'dog', 'frog', 'horse', 'ship', 'truck')
20
 
21
- def inference(input_img, transparency, target_layer_number = -1):
22
  transform = transforms.ToTensor()
23
  input_img = transform(input_img)
24
  input_img = input_img
 
18
  classes = ('plane', 'car', 'bird', 'cat', 'deer',
19
  'dog', 'frog', 'horse', 'ship', 'truck')
20
 
21
+ def inference(input_img, transparency = 0.5, target_layer_number = -1):
22
  transform = transforms.ToTensor()
23
  input_img = transform(input_img)
24
  input_img = input_img