UVIS / requirements.txt
DurgaDeepak's picture
Update requirements.txt
bd6febb verified
raw
history blame
904 Bytes
# Core Libraries
gradio>=3.50 # Web interface for the application
torch>=2.0 # PyTorch for deep learning models
torchvision>=0.15 # TorchVision for pre-trained models and utilities
ultralytics>=8.0 # YOLO models for object detection
opencv-python>=4.7 # OpenCV for video and image processing
# Utility Libraries
numpy>=1.21 # Numerical computations
Pillow>=9.0 # Image processing
requests>=2.28 # HTTP requests for fetching media
timeout-decorator>=0.5.0 # Timeout handling for long-running tasks
tqdm>=4.64 # Progress bars for iterative tasks
# Hugging Face Support
transformers>=4.30 # Hugging Face Transformers for SegFormer models
sentencepiece # Tokenization for Hugging Face models
huggingface-hub>=0.23.0 # Model hub integration for Hugging Face
# Data Handling
pandas>=1.3 # Data manipulation and structured data handling
scipy>=1.7 # Scientific computing for advanced numerical