Update app.py
Browse files
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 |
-
|
49 |
-
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)
|