mshassanali11 commited on
Commit
738df09
Β·
verified Β·
1 Parent(s): 67965f1

Upload 2 files

Browse files
Files changed (2) hide show
  1. packages.txt +2 -3
  2. requirements.txt +9 -24
packages.txt CHANGED
@@ -8,6 +8,5 @@ libgomp1
8
  libopencv-dev
9
  python3-dev
10
  build-essential
11
- libssl-dev # For secure API connections
12
- libffi-dev # For cryptography if needed
13
- ca-certificates # For SSL certificates
 
8
  libopencv-dev
9
  python3-dev
10
  build-essential
11
+ libssl-dev
12
+ ca-certificates
 
requirements.txt CHANGED
@@ -1,8 +1,7 @@
1
  # ─────────────────────────────────────────────
2
  # SmileAI Pro v11 β€” requirements.txt
3
  # ─────────────────────────────────────────────
4
- # Deploy on HuggingFace Spaces β†’ Hardware: T4 GPU
5
- # for ~25s results. CPU works but takes ~3 min.
6
  # ─────────────────────────────────────────────
7
 
8
  # Core UI
@@ -10,40 +9,26 @@ gradio>=4.0.0
10
 
11
  # Image processing
12
  numpy
13
- pillow
14
  opencv-python-headless
15
 
16
- # Face landmark detection (mouth mask precision)
17
  mediapipe==0.10.9
18
 
19
  # PDF export
20
  reportlab
21
 
22
- # ─── Gemini API for Enhanced Teeth Processing ───
23
  google-generativeai>=0.3.0
24
- pillow>=10.0.0 # Already listed, keeping for clarity
25
- python-dotenv>=1.0.0 # For API key management
26
 
27
- # ─── AI Inpainting (GPU) ───────────────────────
28
- # Uncomment the GPU torch line and comment the CPU
29
- # line when deploying on a T4/A10G Space.
30
-
31
- # CPU (default β€” works everywhere, ~3 min/image)
32
  torch
33
  torchvision
34
-
35
- # GPU β€” T4 / A10G HuggingFace Space (~25s/image)
36
- # torch==2.1.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
37
- # torchvision==0.16.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
38
-
39
- # Diffusion pipeline
40
  diffusers>=0.27.0
41
- transformers>=0.27.2 # Updated for better compatibility
42
  accelerate>=0.27.0
43
 
44
- # Memory-efficient attention (GPU only β€” comment out for CPU)
45
- # xformers
46
-
47
  # Utilities
48
- requests>=2.31.0 # For API calls if needed
49
- tqdm>=4.66.0 # Progress bars
 
1
  # ─────────────────────────────────────────────
2
  # SmileAI Pro v11 β€” requirements.txt
3
  # ─────────────────────────────────────────────
4
+ # Simplified for Hugging Face Spaces compatibility
 
5
  # ─────────────────────────────────────────────
6
 
7
  # Core UI
 
9
 
10
  # Image processing
11
  numpy
12
+ pillow>=10.0.0
13
  opencv-python-headless
14
 
15
+ # Face landmark detection
16
  mediapipe==0.10.9
17
 
18
  # PDF export
19
  reportlab
20
 
21
+ # Gemini API
22
  google-generativeai>=0.3.0
23
+ python-dotenv>=1.0.0
 
24
 
25
+ # AI Inpainting - Use CPU version for compatibility
 
 
 
 
26
  torch
27
  torchvision
 
 
 
 
 
 
28
  diffusers>=0.27.0
29
+ transformers>=0.27.2
30
  accelerate>=0.27.0
31
 
 
 
 
32
  # Utilities
33
+ requests>=2.31.0
34
+ tqdm>=4.66.0