Manu101 commited on
Commit
df0fc51
·
verified ·
1 Parent(s): 63c77de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -45,8 +45,8 @@ def inference(input_image, transparency=0.5, target_layer_number=-1):
45
  input_image = np.array(input_image)
46
  org_img = input_image
47
  input_image = input_image.reshape((32, 32, 3))
48
- transforms = transforms.ToTensor()
49
- input_image = transforms(input_image)
50
  input_image = input_image.unsqueeze(0)
51
  outputs = model(input_image)
52
  softmax = torch.nn.Softmax(dim=0)
 
45
  input_image = np.array(input_image)
46
  org_img = input_image
47
  input_image = input_image.reshape((32, 32, 3))
48
+ transform = transforms.ToTensor()
49
+ input_image = transform(input_image)
50
  input_image = input_image.unsqueeze(0)
51
  outputs = model(input_image)
52
  softmax = torch.nn.Softmax(dim=0)