requirements.txt 508 B

123456789101112131415161718192021222324
  1. # requirements.txt
  2. # Core deep learning libraries
  3. torch>=1.9.0
  4. torchvision>=0.10.0
  5. torchaudio>=0.9.0
  6. # For displaying training progress in the command line
  7. tqdm>=4.62.0
  8. # Gradio for the web-based inference UI
  9. gradio>=3.0.0
  10. # Image handling
  11. Pillow>=8.0.0
  12. pickle
  13. # (Optional) if you want to store vocabulary or other metadata in Pickle/JSON, but typically included with Python
  14. # pickle5>=0.0.11
  15. # If you need other libraries (e.g., for advanced tokenization), add them here.
  16. # spacy>=3.0.0
  17. # nltk>=3.6.0