Update requirements.txt

requirements.txt

@@ -1,39 +1,48 @@
-# --- Core Python Packages ---
-numpy
-pillow
-opencv-python
-matplotlib
-requests
-tqdm
-pypdf
-
-# --- Deep Learning ---
-torch>=2.0.0
-tensorflow
-transformers>=4.40.0
-accelerate
-bitsandbytes
-tf-keras
-
-# --- Hugging Face ---
-huggingface-hub
-sentence-transformers
-
-# --- LangChain & RAG ---
-langchain>=0.1.14
-faiss-cpu
-pymupdf
-langchain-community
-
-# --- YOLO Detection ---
-ultralytics
-
-# --- Gradio UI ---
-gradio>=4.28.0
-spaces
-
-# --- Logging & Utility ---
-python-dotenv
-
-mysql
-mysql-connector
+
+--extra-index-url https://download.pytorch.org/whl/cu121
+torch==2.3.1+cu121
+torchvision==0.18.1+cu121
+torchaudio==2.3.1+cu121
+
+# ---------- Core Python Packages ----------
+numpy
+pillow>=10.2
+opencv-python-headless    # headless to avoid X11 issues on Spaces
+matplotlib
+requests
+tqdm
+pypdf
+
+# ---------- Deep Learning ----------
+# TensorFlow MUST be CPU-only so it doesn't probe CUDA in main
+tensorflow-cpu==2.15.*         # pairs well with manylinux + avoids GPU init
+transformers>=4.43,<5          # Qwen2-VL chat pipeline + fast processors
+accelerate>=0.30
+# bitsandbytes often fails on Spaces/ZeroGPU; REMOVE it unless you know you need it
+# bitsandbytes
+
+tf-keras                        # if your code imports tf_keras
+
+# ---------- Hugging Face ----------
+huggingface-hub>=0.23
+sentence-transformers
+
+# ---------- LangChain & RAG ----------
+langchain>=0.1.14
+faiss-cpu
+pymupdf
+langchain-community
+
+# ---------- YOLO Detection ----------
+ultralytics
+
+# ---------- Gradio UI / Spaces ----------
+gradio>=4.28.0
+spaces>=0.19.4
+
+# ---------- Logging & Utility ----------
+python-dotenv
+
+# ---------- Databases ----------
+mysqlclient                  
+mysql-connector-python