Update README.md
Browse files
README.md
CHANGED
|
@@ -58,7 +58,7 @@ from torchvision import transforms
|
|
| 58 |
import numpy as np
|
| 59 |
|
| 60 |
# Load ONNX model
|
| 61 |
-
session = ort.InferenceSession("
|
| 62 |
|
| 63 |
# Get input and output names
|
| 64 |
input_name = session.get_inputs()[0].name
|
|
|
|
| 58 |
import numpy as np
|
| 59 |
|
| 60 |
# Load ONNX model
|
| 61 |
+
session = ort.InferenceSession("AgeRaceGenderNet_v1.onnx")
|
| 62 |
|
| 63 |
# Get input and output names
|
| 64 |
input_name = session.get_inputs()[0].name
|