Update mujtaba_object_detection_code_export.py

mujtaba_object_detection_code_export.py

@@ -1,13 +1,9 @@
-# -*- coding: utf-8 -*-
-"""mujtaba_object_detection_code_export.ipynb
 
-Automatically generated by Colab.
 
-Original file is located at
-    https://colab.research.google.com/drive/1RgE9mbEiNBuPutxI86fIewscERKHyOr_
-"""
 
-!pip install ultralytics datasets wandb gradio opencv-python Pillow captum torchvision --upgrade
+
+# these are the libraries you have to isntall for this projet 
+#!pip install ultralytics datasets wandb gradio opencv-python Pillow captum torchvision --upgrade
 
 import os
 import cv2
@@ -18,7 +14,8 @@ import wandb
 import matplotlib.pyplot as plt
 from datetime import datetime
 from google.colab import userdata
-
+# below is the api key's name I had set in my google colab's secret key name to ensure privacy for my api key . If you want 
+# you can create an accoint at WANDB and then download name the api key WANDB and then use it for your personal use
 wandb.login(key=userdata.get('WANDB'))
 
 def setup_wandb():
@@ -125,7 +122,8 @@ def main():
 
 if __name__ == "__main__":
     main()
-
+# main objective of this colab filw was to generate wegihts after training and thne later downlaod and upload to hugging face for inference
+#as you cannot directly run of hugging due to absence of gpu
 from google.colab import files
 
 files.download("runs/detect/train/weights/best.torchscript")