Update README.md

README.md

@@ -11,3 +11,47 @@ license: mit
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+# The School of AI - ERA(Extensive & Reimagined AI Program) - Assignment 13
+
+This folder consists of Assignment-13 from ERA course offered by - TSAI(The school of AI). 
+Follow https://theschoolof.ai/ for more updates on TSAI
+
+For more details on the assignment, refer to github link: 
+https://github.com/ToletiSri/TSAI_ERA_Assignments/tree/main/S13
+
+As part of the assignment, we have trained a YOLO-V3 model on PASCAL-VOC dataset, using pytorch lightning. PASCAL-VOC dataset is about 5GB of Data. Training this model took about 8 hours of GPU time with 16GB GPU RAM, provided over Kaggle. We have then saved the trained model to file - checkpoint.pth.tar. The saved model is uploaded and used in the current Space. 
+
+As part of the app, we provide provide the following features to the user:
+- allow users to upload new images
+- Provide 3 example images
+
+As part of the output of the app, bounding boxes are shown around the 20 classes, for which the PASCAL VOC dataset is curated. These classes are listed as follows:
+
+    "aeroplane",
+    "bicycle",
+    "bird",
+    "boat",
+    "bottle",
+    "bus",
+    "car",
+    "cat",
+    "chair",
+    "cow",
+    "diningtable",
+    "dog",
+    "horse",
+    "motorbike",
+    "person",
+    "pottedplant",
+    "sheep",
+    "sofa",
+    "train",
+    "tvmonitor"
+
+
+
+
+---
+
+