PJ2005 commited on
Commit
00065ee
·
verified ·
1 Parent(s): 9daa7e5

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -1,3 +1,5 @@
 
 
1
  gradio
2
  transformers
3
  pandas
@@ -6,4 +8,3 @@ scikit-learn
6
  plotly
7
  emoji
8
  langdetect
9
- torch
 
1
+ torch==2.0.1
2
+ torchvision==0.15.2 # optional if using image tasks
3
  gradio
4
  transformers
5
  pandas
 
8
  plotly
9
  emoji
10
  langdetect