Spaces:
Running
Running
RizwanMunawar
commited on
Commit
•
8ff935e
1
Parent(s):
e42a8c2
fix torch and torchvision dependencies
Browse files- requirements.txt +4 -1
requirements.txt
CHANGED
|
@@ -1,3 +1,6 @@
|
|
| 1 |
gradio
|
|
|
|
|
|
|
|
|
|
| 2 |
ultralytics
|
| 3 |
-
Pillow
|
|
|
|
| 1 |
gradio
|
| 2 |
+
torch>=1.8.0
|
| 3 |
+
torch>=1.8.0,!=2.4.0; sys_platform == 'win32' # Windows CPU errors w/ 2.4.0 https://github.com/ultralytics/ultralytics/issues/15049
|
| 4 |
+
torchvision>=0.9.0
|
| 5 |
ultralytics
|
| 6 |
+
Pillow
|