fastai
classification
waste
resnet
santit96 commited on
Commit
e6b7009
·
1 Parent(s): b67ede0

update readme with model name and demo information

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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]:** [TODO]()
46
 
47
  ## Uses
48
 
49
- At Rootstrap we classify waste. We found that people were straggled 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,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(f'result-resnet34.pkl')
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