vlad-m-dev commited on
Commit
c8dd36b
·
verified ·
1 Parent(s): f566f88

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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