Update app.py
Browse files
app.py
CHANGED
|
@@ -19,6 +19,7 @@ model.load_state_dict(torch.load("model.pth", map_location=torch.device('cpu')),
|
|
| 19 |
|
| 20 |
def inference(input_img, transparency = 0.5, target_layer_number = -1):
|
| 21 |
input_img = np.array(Image.fromarray(np.array(input_img)).resize((32,32)))
|
|
|
|
| 22 |
transform = transforms.ToTensor()
|
| 23 |
input_img = transform(input_img).unsqueeze(0)
|
| 24 |
outputs = model(input_img)
|
|
|
|
| 19 |
|
| 20 |
def inference(input_img, transparency = 0.5, target_layer_number = -1):
|
| 21 |
input_img = np.array(Image.fromarray(np.array(input_img)).resize((32,32)))
|
| 22 |
+
org_img = input_img
|
| 23 |
transform = transforms.ToTensor()
|
| 24 |
input_img = transform(input_img).unsqueeze(0)
|
| 25 |
outputs = model(input_img)
|