Update README.md

README.md

@@ -1,3 +1,19 @@
----
-license: mit
----
+---
+license: mit
+---
+
+## Usage
+
+```python
+state_dict = torch.load('model.pth', map_location='cpu')
+model.load_state_dict(state_dict, strict=False)
+model.eval()
+
+def predict(im):
+    x = torch.tensor(im, dtype=torch.float32)
+
+    with torch.no_grad():
+        out = model(x)
+
+    probabilities = F.softmax(out[0], dim=0)
+```