Spaces:
Runtime error
Runtime error
update
Browse files
app.py
CHANGED
|
@@ -42,8 +42,8 @@ sub_test_list = [i.strip() for i in sub_test_list]
|
|
| 42 |
embeddings = torch.load('embeddings.pt')
|
| 43 |
# Configure
|
| 44 |
options = onnxruntime.SessionOptions()
|
| 45 |
-
options.intra_op_num_threads =
|
| 46 |
-
options.inter_op_num_threads =
|
| 47 |
# Load model
|
| 48 |
PATH = 'model_onnx.onnx'
|
| 49 |
ort_session = onnxruntime.InferenceSession(PATH)
|
|
|
|
| 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)
|