update

.idea/.gitignore

@@ -0,0 +1,5 @@
+# Default ignored files
+/shelf/
+/workspace.xml
+*.xml
+*.iml

app.py

@@ -40,6 +40,10 @@ sub_test_list = open('img_list.txt', 'r')
 sub_test_list = [i.strip() for i in sub_test_list]
 # Load images embedding for similarity
 embeddings = torch.load('embeddings.pt')
+# Configure
+options = onnxruntime.SessionOptions()
+options.intra_op_num_threads = 2
+options.inter_op_num_threads = 2
 # Load model
 PATH = 'model_onnx.onnx'
 ort_session = onnxruntime.InferenceSession(PATH)