Merge pull request #55 from duggalsu/mem_profiling

Add memory profiling

src/api/.gitignore

@@ -1 +1,3 @@
-credentials.json
+credentials.json
+*.bin
+*.html

src/api/core/operators/vid_vec_rep_resnet.py

@@ -1,7 +1,7 @@
 
 
 def initialize(param):
-    print("Intalling packages for vid_vec_rep_resnet")
+    print("Installing packages for vid_vec_rep_resnet")
 
     global TextBlob, os, np, cv2, qr, torch, data, models, transforms, Image, FFmpeg, wget
     global imagenet_transform, ImageListDataset, VideoAnalyzer, gendata, compress_video
@@ -247,4 +247,4 @@ def state():
 #         #     print(len(res))
 #         print("LEN - ", len(result_list[0]["vid_vec"]))
 #     except Exception as e:
-#         print(f"Error: {e}")
+#         print(f"Error: {e}")

src/api/requirements.in

@@ -20,4 +20,5 @@ click==8.1.7
 typing-extensions==4.9.0
 pytest==7.4.4
 PyYAML==6.0.1
-dacite==1.8.1
+dacite==1.8.1
+memray==1.11.0  # dev

src/api/requirements.txt

@@ -2,7 +2,7 @@
 # This file is autogenerated by pip-compile with Python 3.11
 # by the following command:
 #
-#    pip-compile --find-links=https://download.pytorch.org/whl/torch_stable.html
+#    pip-compile --find-links=https://download.pytorch.org/whl/torch_stable.html requirements.in
 #
 --find-links https://download.pytorch.org/whl/torch_stable.html
 
@@ -86,6 +86,7 @@ itsdangerous==2.1.2
 jinja2==3.1.3
     # via
     #   flask
+    #   memray
     #   torch
 jmespath==0.10.0
     # via
@@ -97,10 +98,23 @@ joblib==1.3.2
     #   scikit-learn
 lazy-loader==0.3
     # via scikit-image
+linkify-it-py==2.0.3
+    # via markdown-it-py
+markdown-it-py[linkify,plugins]==3.0.0
+    # via
+    #   mdit-py-plugins
+    #   rich
+    #   textual
 markupsafe==2.1.4
     # via
     #   jinja2
     #   werkzeug
+mdit-py-plugins==0.4.0
+    # via markdown-it-py
+mdurl==0.1.2
+    # via markdown-it-py
+memray==1.11.0
+    # via -r requirements.in
 mpmath==1.3.0
     # via sympy
 networkx==3.2.1
@@ -155,6 +169,8 @@ pyasn1==0.4.8
     #   rsa
 pyasn1-modules==0.2.8
     # via google-auth
+pygments==2.17.2
+    # via rich
 pytest==7.4.4
     # via -r requirements.in
 python-dateutil==2.8.1
@@ -176,6 +192,10 @@ requests==2.31.0
     #   huggingface-hub
     #   torchvision
     #   transformers
+rich==13.7.0
+    # via
+    #   memray
+    #   textual
 rsa==4.7.2
     # via google-auth
 s3transfer==0.10.0
@@ -204,6 +224,8 @@ sympy==1.12
     # via torch
 textblob==0.17.1
     # via -r requirements.in
+textual==0.48.2
+    # via memray
 threadpoolctl==2.1.0
     # via scikit-learn
 tifffile==2023.12.9
@@ -232,7 +254,10 @@ typing-extensions==4.9.0
     # via
     #   -r requirements.in
     #   huggingface-hub
+    #   textual
     #   torch
+uc-micro-py==1.0.2
+    # via linkify-it-py
 urllib3==2.0.7
     # via
     #   botocore