muskangoyal06 commited on
Commit
8ffc99f
·
verified ·
1 Parent(s): 4557350

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -3,5 +3,5 @@ ultralytics==8.0.43
3
  gradio==3.50.0
4
  pillow==10.0.0
5
  numpy==1.23.5
6
- torch==2.0.1
7
- torchvision==0.15.2
 
3
  gradio==3.50.0
4
  pillow==10.0.0
5
  numpy==1.23.5
6
+ torch==2.0.1+cu117
7
+ torchvision==0.15.2+cu117