Update app.py
Browse files
app.py
CHANGED
@@ -100,8 +100,6 @@ def inference(img):
|
|
100 |
input_image = Image.open(img)
|
101 |
orig_tensor = np.asarray(input_image)
|
102 |
input_tensor = preprocess(input_image)
|
103 |
-
input_tensor = input_tensor.unsqueeze(0)
|
104 |
-
input_tensor = input_tensor.detach().cpu().numpy()
|
105 |
output_names = list(map(lambda output: output.name, outputs))
|
106 |
input_name = sess.get_inputs()[0].name
|
107 |
detections = sess.run(output_names, {input_name: input_tensor})
|
|
|
100 |
input_image = Image.open(img)
|
101 |
orig_tensor = np.asarray(input_image)
|
102 |
input_tensor = preprocess(input_image)
|
|
|
|
|
103 |
output_names = list(map(lambda output: output.name, outputs))
|
104 |
input_name = sess.get_inputs()[0].name
|
105 |
detections = sess.run(output_names, {input_name: input_tensor})
|