behtarino / requirments.txt
amirhosseinmoayedi's picture
Update requirments.txt
492f7ba verified
raw
history blame contribute delete
409 Bytes
transformers==4.34.0 # Hugging Face library for models and tokenizers
torch>=1.12.0 # PyTorch for running models (CPU or GPU)
sentencepiece==0.1.99 # For models that use SentencePiece tokenizers
gradio==3.41.0 # For creating a web interface (optional but useful)
fastapi==0.95.2 # For building APIs (optional if using FastAPI)
uvicorn==0.22.0 # For serving FastAPI apps (optional)