Update README.md
Browse files
README.md
CHANGED
@@ -92,7 +92,7 @@ session = ort.InferenceSession("mobilenetv2_doc_photo_quant.onnx")
|
|
92 |
img = np.random.randn(1, 3, 224, 224).astype(np.float32) # Replace with your image preprocessing!
|
93 |
output = session.run(None, {"input": img})
|
94 |
pred_class = np.argmax(output[0])
|
95 |
-
print(pred_class) # 0 = document, 1 = photo
|
96 |
|
97 |
---
|
98 |
|
|
|
92 |
img = np.random.randn(1, 3, 224, 224).astype(np.float32) # Replace with your image preprocessing!
|
93 |
output = session.run(None, {"input": img})
|
94 |
pred_class = np.argmax(output[0])
|
95 |
+
print(pred_class) # 0 = document, 1 = photo```
|
96 |
|
97 |
---
|
98 |
|