update readme with model name and demo information
Browse files
README.md
CHANGED
@@ -42,11 +42,11 @@ To be also configurable.
|
|
42 |
### Model Sources
|
43 |
|
44 |
- **Repository:** [https://github.com/rootstrap/fastai-waste-classifier](https://github.com/rootstrap/fastai-waste-classifier)
|
45 |
-
- **Demo [optional]:** [
|
46 |
|
47 |
## Uses
|
48 |
|
49 |
-
At Rootstrap we classify waste. We found that people were
|
50 |
and then we end up not recycling most of the generated waste at the office, since if there were items in the wrong basket,
|
51 |
all the basket should not be classified.
|
52 |
Because of this reason, we created an app to help people at our company to classify waste.
|
@@ -54,7 +54,7 @@ Because of this reason, we created an app to help people at our company to class
|
|
54 |
### Direct Use
|
55 |
|
56 |
```bash
|
57 |
-
model = load_learner(
|
58 |
model.predict()
|
59 |
```
|
60 |
|
|
|
42 |
### Model Sources
|
43 |
|
44 |
- **Repository:** [https://github.com/rootstrap/fastai-waste-classifier](https://github.com/rootstrap/fastai-waste-classifier)
|
45 |
+
- **Demo [optional]:** [https://huggingface.co/spaces/rootstrap-org/waste-classifier](https://huggingface.co/spaces/rootstrap-org/waste-classifier)
|
46 |
|
47 |
## Uses
|
48 |
|
49 |
+
At Rootstrap we classify waste. We found that people were struggled to classify correctly,
|
50 |
and then we end up not recycling most of the generated waste at the office, since if there were items in the wrong basket,
|
51 |
all the basket should not be classified.
|
52 |
Because of this reason, we created an app to help people at our company to classify waste.
|
|
|
54 |
### Direct Use
|
55 |
|
56 |
```bash
|
57 |
+
model = load_learner("result-resnet50.pkl")
|
58 |
model.predict()
|
59 |
```
|
60 |
|