Updated Requirements file

requirements.txt

@@ -1,20 +1,40 @@
-torch
-torchvision
-Pillow
-numpy
-matplotlib
-tqdm
-nltk
-rouge-score
-pycocotools
-pycocoevalcap @ git+https://github.com/salaniz/pycocoevalcap.git
-flask
-flask_sqlalchemy
-gunicorn
-ultralytics
+
+# Core PyTorch (use CPU-only versions for faster install)
+torch==2.0.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
+torchvision==0.15.2+cpu -f https://download.pytorch.org/whl/torch_stable.html
+
+# Image processing
+Pillow==9.5.0
+opencv-python-headless==4.8.0.76
+
+# Core ML libraries
+numpy==1.24.3
+matplotlib==3.7.1
+
+# NLP libraries
+nltk==3.8.1
+rouge-score==0.1.2
+
+# COCO evaluation (this is problematic - see alternatives below)
+# pycocotools==2.0.6
+# pycocoevalcap @ git+https://github.com/salaniz/pycocoevalcap.git
+
+# Web framework
+flask==2.3.2
+flask-sqlalchemy==3.0.5
+gunicorn==21.2.0
+werkzeug==2.3.6
+
+# Object detection
+ultralytics==8.0.120
+
+# Face recognition - LIGHTER ALTERNATIVES
+# Option 1: MediaPipe (recommended - much faster)
+mediapipe==0.10.1
+
+# Option 2: If you must use dlib-based face recognition
+face-recognition==1.3.0
 dlib-bin==19.22.0
-face_recognition==1.3.0
-Pillow
-opencv-python-headless
-numpy
-werkzeug
+
+# Progress tracking
+tqdm==4.65.0