sumitsingh830 commited on
Commit
10f8ce7
·
verified ·
1 Parent(s): 23f6e20

Upload 2 files

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +18 -2
Dockerfile CHANGED
@@ -19,7 +19,7 @@ RUN apt-get update && apt-get install -y \
19
  WORKDIR /app
20
 
21
  # Copy requirements first for better caching
22
- COPY requirements_hf.txt requirements.txt
23
 
24
  # Install Python dependencies (excluding sam2, which will be installed from local directory)
25
  # This includes hydra-core which is required by sam2
 
19
  WORKDIR /app
20
 
21
  # Copy requirements first for better caching
22
+ COPY requirements.txt requirements.txt
23
 
24
  # Install Python dependencies (excluding sam2, which will be installed from local directory)
25
  # This includes hydra-core which is required by sam2
requirements.txt CHANGED
@@ -1,11 +1,27 @@
 
 
1
  fastapi>=0.104.1
2
  uvicorn[standard]>=0.24.0
 
 
3
  numpy>=1.26.0
4
  opencv-python>=4.8.0
5
- requests>=2.31.0
6
  scikit-image>=0.21.0
 
 
 
7
  torch>=2.0.0
8
  torchvision>=0.15.0
9
- segment-anything
10
  huggingface_hub>=0.20.0
 
 
 
 
 
 
 
 
 
 
11
  psutil>=5.9.0
 
1
+ # Requirements for Hugging Face Spaces deployment with GPU support
2
+ # FastAPI and server
3
  fastapi>=0.104.1
4
  uvicorn[standard]>=0.24.0
5
+
6
+ # Image processing
7
  numpy>=1.26.0
8
  opencv-python>=4.8.0
9
+ Pillow>=10.0.0
10
  scikit-image>=0.21.0
11
+
12
+ # Deep learning and SAM2
13
+ # Note: sam2 package will be installed from local directory in Dockerfile
14
  torch>=2.0.0
15
  torchvision>=0.15.0
 
16
  huggingface_hub>=0.20.0
17
+
18
+ # SAM2 dependencies (required for sam2 package)
19
+ # Note: omegaconf is a dependency of hydra-core but explicitly listed for safety
20
+ omegaconf>=2.3.0
21
+ hydra-core>=1.3.2
22
+ iopath>=0.1.10
23
+ tqdm>=4.66.1
24
+
25
+ # Utilities
26
+ requests>=2.31.0
27
  psutil>=5.9.0