Update README.md

README.md

@@ -1,3 +1,52 @@
----
-license: unknown
----
+---
+license: unknown
+metrics:
+- accuracy
+pipeline_tag: image-classification
+tags:
+- fin
+- image
+---
+
+
+## Download model
+
+```bash
+git lfs install
+git clone https://huggingface.co/Abner0803/multiclass-stock-cnn
+```
+
+## Usage
+
+In directory `./multiclass-stock-cnn', you can try the pre-trained model out
+
+```bash
+touch expr.py
+
+pip install torch pillow numpy
+```
+
+```python
+## expr.py
+
+import torch
+import numpy as np
+from PIL import Image
+
+
+image_file = "temp.png"
+matrix = np.array(Image.open(image_file))
+tensor = torch.Tensor(matrix, dtype=torch.float32)
+
+pred = model(tensor.unsqueeze(0).unsqueeze(0))
+
+print(pred)
+```
+
+You'll get the probability of each class
+
+```bash
+tensor([[9.9691e-01, 1.1429e-09, 3.4367e-20, 6.5349e-30, 9.6665e-25, 1.1544e-27,
+         2.8234e-11, 6.8401e-21, 2.1943e-16, 3.0882e-03]],
+       grad_fn=<SoftmaxBackward0>)
+```