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)
|