starrednt2 / requirements.txt
Futi613's picture
Update requirements.txt
b410914 verified
# --- Deep Learning Core ---
torch==2.2.2+cpu
torchvision==0.17.2+cpu
onnx==1.15.0
onnxruntime==1.17.3
# --- NumPy & image handling ---
numpy==1.24.4
Pillow==10.2.0
# --- Protobuf (fixes ONNX serialization issues) ---
protobuf==3.20.3
# --- Gradio / Interface ---
gradio==4.44.0
# --- Optional (recommandé pour compatibilité PyTorch / ONNX) ---
typing-extensions>=4.9.0
# --- Source for CPU wheels ---
-f https://download.pytorch.org/whl/torch_stable.html