torch
transformers
uvicorn==0.15.0
streamlit