Hugging Face's logo

Spaces:
Skym616
/
tumor-prediction
Running

App Files Community
Skym616 commited on
Commit
fe7a354
·
1 Parent(s): 93f30c6

fix cudda

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -3,5 +3,5 @@ Pillow==11.1.0
3
  python-dotenv==1.0.1
4
  Requests==2.32.3
5
  streamlit==1.42.0
6
- torch==2.6.0+cu126
7
- torchvision==0.21.0+cu126
 
3
  python-dotenv==1.0.1
4
  Requests==2.32.3
5
  streamlit==1.42.0
6
+ torch==2.6.0
7
+ torchvision==0.21.0