Spaces:
Build error
Build error
Update README.md
Browse files
README.md
CHANGED
@@ -31,3 +31,7 @@ To make a prediction, the model will be available for interaction in the Hugging
|
|
31 |
Building models
|
32 |
The application is based on a single model, based on a pre-trained ANN, ResNet50. Additional layers have been added for the purpose of dog breed classification and the pre-trained layers were frozen.
|
33 |
|
|
|
|
|
|
|
|
|
|
31 |
Building models
|
32 |
The application is based on a single model, based on a pre-trained ANN, ResNet50. Additional layers have been added for the purpose of dog breed classification and the pre-trained layers were frozen.
|
33 |
|
34 |
+
This repository has Three main files:
|
35 |
+
- notebook.ipynb: This notebook contains all the steps that we took to create the RestNet50 Classification Model.
|
36 |
+
- app.py: This file contains the Gradio UI and also loads the model to HugginFace.
|
37 |
+
- model.zip: The model itself, already trained on the notebook and ready to be loaded.
|