Update README.md

README.md

@@ -1,3 +1,24 @@
----
-license: apache-2.0
----
+---
+license: apache-2.0
+base_model:
+- google/efficientnet-b2
+---
+
+## Overview
+
+Detects bad crop jobs and misalignments, like screenshots from phones that should be trimmed down to relevant details.
+
+Version 8.
+
+Training progress: https://wandb.ai/josephc/bad_crop/overview
+
+## Usage:
+
+```python
+import numpy
+import onnxruntime as ort
+session = ort.InferenceSession("./bad_crop.onnx")
+blank = numpy.zeros((1, 3, 224, 224), dtype=numpy.float32)
+out = torch.softmax(torch.tensor(session.run(["output"], {"input": blank,})[0][0]), axis=0)
+print(out)  # tensor([9.9941e-01, 5.8832e-04])  # Bad crop, good crop
+```