Update README.md

README.md

@@ -42,7 +42,7 @@ To be also configurable.
 ### Model Sources 
 
 - **Repository:** [https://github.com/rootstrap/fastai-waste-classifier](https://github.com/rootstrap/fastai-waste-classifier)
-- **Demo [optional]:** [https://huggingface.co/spaces/rootstrap-org/waste-classifier](https://huggingface.co/spaces/rootstrap-org/waste-classifier)
+- **Demo (with detector):** [https://huggingface.co/spaces/rootstrap-org/waste-classifier](https://huggingface.co/spaces/rootstrap-org/waste-classifier)
 
 ## Uses
 
@@ -76,8 +76,8 @@ You can find the code used to download the images at [this](https://colab.resear
 You can find the final dataset splitted into train and test  [rootstrap-org/waste-classifier](https://huggingface.co/datasets/rootstrap-org/waste-classifier) in a Google Drive folder.
 
 Split into train/test 
-You can find the code for the classifier [resnet-model.ipynb](resnet-model.ipynb)
-To load the model in the file [result-resnet34.pkl](result-resnet34.pkl)
+You can find the code for the classifier [resnet-model.ipynb](https://github.com/rootstrap/fastai-waste-classifier/blob/main/resnet-model.ipynb)
+To load the model in the file [result-resnet50.pkl](result-resnet50.pkl)
 
 The model has been trained to classify waste into 6 classes. 
 This repository contains the dataset that we collected.