| # Base libraries for Hugging Face models | |
| transformers==4.30.0 # Hugging Face Transformers library for model management | |
| torch==2.0.0 # PyTorch library for model training and inference | |
| einops==0.6.0 # Einops for tensor operations and manipulation | |
| # or for TensorFlow | |
| # tensorflow==2.12.0 # Uncomment if using TensorFlow | |
| # Other libraries commonly used in machine learning projects | |
| numpy==1.24.0 # NumPy for numerical operations | |
| pandas==1.5.3 # Pandas for data manipulation | |
| scikit-learn==1.2.0 # Scikit-learn for machine learning utilities | |
| datasets==2.12.0 # Hugging Face Datasets library for dataset handling | |
| # Add other libraries as needed | |
| # Web app dependencies | |
| fastapi==0.95.0 # FastAPI for building the API | |
| uvicorn==0.22.0 # Uvicorn for serving the FastAPI app | |
| pydantic==1.11.1 # Pydantic for data validation | |
| # Optional: For logging and monitoring (if used in your project) | |
| wandb==0.15.0 # Weights & Biases for experiment tracking | |