sanabanu31 commited on
Commit
0f8b2cc
·
verified ·
1 Parent(s): 9018f12

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -1,8 +1,9 @@
1
  fastapi
2
  uvicorn
3
- transformers
4
  sentencepiece
5
  scikit-learn
6
  pandas
7
  numpy
8
  torch
 
 
1
  fastapi
2
  uvicorn
3
+ transformers[torch]
4
  sentencepiece
5
  scikit-learn
6
  pandas
7
  numpy
8
  torch
9
+ protobuf