Alessio Grancini commited on
Commit
865e874
·
verified ·
1 Parent(s): 53cb09e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -5
requirements.txt CHANGED
@@ -11,10 +11,10 @@ urllib3>=1.26.14
11
 
12
  # ZeroGPU specific requirements
13
  --extra-index-url https://download.pytorch.org/whl/cu118
14
- torch==2.0.1+cu118 # ZeroGPU compatible version
15
- torchvision==0.15.2+cu118 # Compatible with torch 2.0.1
16
- gradio>=4.0.0 # Required for ZeroGPU
17
- spaces # Required for ZeroGPU
18
 
19
  # Model specific
20
  ultralytics>=8.0.132
@@ -22,7 +22,7 @@ timm==0.6.12
22
 
23
  # Visualization
24
  open3d==0.16.0
25
- plotly # For 3D visualization
26
  pandas>=1.1.4
27
  seaborn>=0.11.0
28
 
 
11
 
12
  # ZeroGPU specific requirements
13
  --extra-index-url https://download.pytorch.org/whl/cu118
14
+ torch==2.0.1+cu118
15
+ torchvision==0.15.2+cu118
16
+ gradio==4.44.1 # Specific version for ZeroGPU compatibility
17
+ spaces
18
 
19
  # Model specific
20
  ultralytics>=8.0.132
 
22
 
23
  # Visualization
24
  open3d==0.16.0
25
+ plotly
26
  pandas>=1.1.4
27
  seaborn>=0.11.0
28