fj11 commited on
Commit
0c56e2e
·
verified ·
1 Parent(s): 8b653cc

Create requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +23 -0
requirements.txt ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # ===== Core (Gradio inference app) =====
2
+ gradio>=4.32.0
3
+ transformers>=4.41.0
4
+ torch>=2.3.0 # For CUDA, install from https://pytorch.org/get-started/ per your GPU/driver
5
+ safetensors>=0.4.3
6
+ huggingface_hub>=0.23.0
7
+ numpy>=1.24.0
8
+ pillow>=10.0.0
9
+
10
+ # ===== Training & dataset pipeline (optional but handy if you also train or preprocess) =====
11
+ accelerate>=0.30.0
12
+ datasets>=2.20.0
13
+ pyarrow>=15.0.0
14
+ evaluate>=0.4.2
15
+ iterative-stratification>=0.1.7
16
+ matplotlib>=3.8.0
17
+ tensorboard>=2.16.0
18
+ opencv-python>=4.8.0
19
+ scikit-learn>=1.3.0
20
+ scipy>=1.10.0
21
+
22
+ # ===== (Optional) convenience =====
23
+ # streamlit>=1.33.0 # only if you still run a Streamlit demo