Update README.md

README.md

@@ -20,7 +20,7 @@ The output of the application is a text describing the breed of the dog or the c
 The task the application will be performing is classification. Specifically, the classification of images of dogs, with the objective of accurately classifying them according to their breed. As such, the input will be images of dogs, and the possible values for the output will be one of one hundred and twenty two different breeds; e.g. Poodle, Briard, Newfoundland, Bluetick, etc.
 
 # Data collection
-The initial train dataset will be the Stanford Dogs dataset. The Stanford Dogs dataset contains images of 120 breeds of dogs from around the world. This dataset has been built using images and annotation from ImageNet for the task of fine-grained image categorization[1]. Contents of this dataset:
+The initial train dataset will be the Stanford Dogs dataset. The Stanford Dogs dataset contains images of 120 breeds of dogs from around the world. This dataset has been built using images and annotation from ImageNet for the task of fine-grained image categorization. Contents of this dataset:
 
 Number of categories: 120
 Number of images: 20,580
@@ -28,7 +28,8 @@ Annotations: Class labels, Bounding boxes
 
 # Making Predictions
 To make a prediction, the model will be available for interaction in the HuggingFace platform, over at https://huggingface.co/spaces/Samood/whos_that_doggo
-Building models 
+
+# Building models 
 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.
 
 This repository has Three main files: