FrancescoLR commited on
Commit
46541b6
·
1 Parent(s): 18f25b9

Updated app.py

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -1
requirements.txt CHANGED
@@ -1,5 +1,7 @@
1
  gradio
2
- torch
 
 
3
  nnunetv2
4
  nibabel
5
  numpy
 
1
  gradio
2
+ torch==2.0.1+cu118
3
+ torchvision==0.15.2+cu118
4
+ torchaudio==2.0.2+cu118
5
  nnunetv2
6
  nibabel
7
  numpy