DawnC commited on
Commit
a0176e2
1 Parent(s): f0b6beb

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -3
requirements.txt CHANGED
@@ -1,6 +1,5 @@
1
-
2
- torch==2.4.1+cu118
3
- torchvision==0.19.1+cu118
4
  gradio==5.6.0
5
  Pillow==10.4.0
6
  ultralytics
 
1
+ torch==2.5.1+cu121
2
+ torchvision==0.20.1+cu121
 
3
  gradio==5.6.0
4
  Pillow==10.4.0
5
  ultralytics