Mountchicken commited on
Commit
054f314
·
verified ·
1 Parent(s): bdabc21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import argparse
2
  import json
3
  import os
 
4
  import sys
5
  import threading
6
 
 
1
  import argparse
2
  import json
3
  import os
4
+ os.system("pip install torch==2.6.0 torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu124")
5
  import sys
6
  import threading
7