Hugging Face's logo

Spaces:
MingGatsby
/
Streamlit_Dicom_CLOTS_Classification
Sleeping

App Files Community
MingGatsby commited on
Commit
e896750
•
1 Parent(s): 67cf200

Update requirement.txt

Browse files
Files changed (1) hide show
  1. requirement.txt +3 -1
requirement.txt CHANGED
@@ -1,4 +1,6 @@
1
  streamlit
2
  gradio
3
  monai
4
- torch==2.0.1
 
 
 
1
  streamlit
2
  gradio
3
  monai
4
+ torch==2.0.1 --index-url https://download.pytorch.org/whl/cu117
5
+ torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu117
6
+ torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu117