vumichien commited on
Commit
9139749
1 Parent(s): 8fb55c9
Files changed (1) hide show
  1. app.py +4 -0
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)