classification / requirements.txt
rajkumarrawal's picture
refactor: simplify model device handling using accelerate library
826cc00
Raw
History Blame Contribute Delete
118 Bytes
transformers
torch
requests
Pillow
open_clip_torch
ftfy
accelerate
# This is only needed for local deployment
gradio