hamxaameer commited on
Commit
27ddb97
·
verified ·
1 Parent(s): 38690d3

Create requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +22 -0
requirements.txt ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # MAE - Masked Autoencoder Requirements
2
+ # For Hugging Face Spaces Deployment
3
+
4
+ # Core PyTorch (CPU version for HF free tier, auto-detects GPU)
5
+ torch>=2.0.0
6
+ torchvision>=0.15.0
7
+
8
+ # Image processing
9
+ numpy>=1.24.0
10
+ pillow>=9.5.0
11
+
12
+ # Tensor operations (required for model)
13
+ einops>=0.6.0
14
+
15
+ # Gradio for web interface
16
+ gradio>=4.0.0
17
+
18
+ # Visualization (optional, for metrics)
19
+ matplotlib>=3.7.0
20
+
21
+ # Progress bars
22
+ tqdm>=4.65.0