Sanjayraju30 commited on
Commit
400b65c
·
verified ·
1 Parent(s): 6993b45

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,7 +1,8 @@
1
  gradio
2
  transformers
3
- torch
 
4
  Pillow
5
  numpy
6
  opencv-python-headless
7
- timm # ✅ Required for DETR model
 
1
  gradio
2
  transformers
3
+ torch==2.2.2+cpu
4
+ --extra-index-url https://download.pytorch.org/whl/cpu
5
  Pillow
6
  numpy
7
  opencv-python-headless
8
+ timm