angusfung's picture
Initial setup with Longformer embedding feature
7812756
raw
history blame contribute delete
317 Bytes
torch>=1.13.0
numpy>=1.22.0
transformers>=4.26.0
gensim>=4.3.0
fastapi>=0.95.0
uvicorn>=0.22.0
shap>=0.41.0
pydantic>=1.10.7
sentence-transformers>=2.2.2
# Specific models needed
tokenizers>=0.13.2
sentencepiece>=0.1.97
# For text processing
nltk>=3.7
# For web requests
requests>=2.28.0
# For CORS
starlette>=0.26.0