Update requirements.txt

requirements.txt

@@ -1,46 +1,42 @@
+# ---- PyTorch (pick a ZeroGPU-supported set; 2.3.x is NOT allowed) ----
+# Option A (recommended today): Torch 2.4 line
+torch==2.4.0
+torchvision==0.19.0
+torchaudio==2.4.0
 
---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 ----------
+# ---- Core Python Packages ----
 numpy
 pillow>=10.2
-opencv-python-headless    # headless to avoid X11 issues on Spaces
+opencv-python-headless
 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
+# ---- Deep Learning / HF ----
+# Keep TF on CPU so it doesn't probe CUDA in the main process
+tensorflow-cpu==2.15.*
+transformers>=4.43,<5
 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 ----------
+# bitsandbytes  # avoid on ZeroGPU unless you *really* know you need it
+tf-keras
 huggingface-hub>=0.23
 sentence-transformers
 
-# ---------- LangChain & RAG ----------
+# ---- LangChain & RAG ----
 langchain>=0.1.14
 faiss-cpu
 pymupdf
 langchain-community
 
-# ---------- YOLO Detection ----------
+# ---- YOLO Detection ----
 ultralytics
 
-# ---------- Gradio UI / Spaces ----------
+# ---- Gradio UI / Spaces ----
 gradio>=4.28.0
 spaces>=0.19.4
 
-# ---------- Logging & Utility ----------
+# ---- Logging & Utility ----
 python-dotenv
 
 # ---------- Databases ----------