Spaces:
Running
Running
update
Browse files
app.py
CHANGED
|
@@ -35,15 +35,19 @@ def transform_image(myarray):
|
|
| 35 |
f = open('imagenet_label.json',)
|
| 36 |
label_map=json.load(f)
|
| 37 |
f.close()
|
|
|
|
| 38 |
# Load list of images for similarity
|
| 39 |
sub_test_list = open('img_list.txt', 'r')
|
| 40 |
sub_test_list = [i.strip() for i in sub_test_list]
|
|
|
|
| 41 |
# Load images embedding for similarity
|
| 42 |
embeddings = torch.load('embeddings.pt')
|
|
|
|
| 43 |
# Configure
|
| 44 |
options = onnxruntime.SessionOptions()
|
| 45 |
options.intra_op_num_threads = 1
|
| 46 |
options.inter_op_num_threads = 1
|
|
|
|
| 47 |
# Load model
|
| 48 |
PATH = 'model_onnx.onnx'
|
| 49 |
ort_session = onnxruntime.InferenceSession(PATH)
|
|
|
|
| 35 |
f = open('imagenet_label.json',)
|
| 36 |
label_map=json.load(f)
|
| 37 |
f.close()
|
| 38 |
+
|
| 39 |
# Load list of images for similarity
|
| 40 |
sub_test_list = open('img_list.txt', 'r')
|
| 41 |
sub_test_list = [i.strip() for i in sub_test_list]
|
| 42 |
+
|
| 43 |
# Load images embedding for similarity
|
| 44 |
embeddings = torch.load('embeddings.pt')
|
| 45 |
+
|
| 46 |
# Configure
|
| 47 |
options = onnxruntime.SessionOptions()
|
| 48 |
options.intra_op_num_threads = 1
|
| 49 |
options.inter_op_num_threads = 1
|
| 50 |
+
|
| 51 |
# Load model
|
| 52 |
PATH = 'model_onnx.onnx'
|
| 53 |
ort_session = onnxruntime.InferenceSession(PATH)
|