Hugging Face's logo

Spaces:
dylanplummer
/
NextJump
Runtime error

App Files Community
dylanplummer commited on
Commit
f2f4e48
·
1 Parent(s): 796d94e

testing cuda

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +4 -3
app.py CHANGED
@@ -29,7 +29,7 @@ plt.style.use('dark_background')
29
 
30
  checkpoint = hf_hub_download(repo_id="dylanplummer/ropenet", filename="ropenet_keypoint_0.pt", repo_type="model", token=os.environ['DATASET_SECRET'])
31
  model_file = checkpoint = hf_hub_download(repo_id="dylanplummer/ropenet", filename="model.py", repo_type="model", token=os.environ['DATASET_SECRET'])
32
- os.move(model_file, "model.py")
33
  from model import RepNet
34
  # model_xml = hf_hub_download(repo_id="dylanplummer/ropenet", filename="model.xml", repo_type="model", token=os.environ['DATASET_SECRET'])
35
  # hf_hub_download(repo_id="dylanplummer/ropenet", filename="model.mapping", repo_type="model", token=os.environ['DATASET_SECRET'])
 
29
 
30
  checkpoint = hf_hub_download(repo_id="dylanplummer/ropenet", filename="ropenet_keypoint_0.pt", repo_type="model", token=os.environ['DATASET_SECRET'])
31
  model_file = checkpoint = hf_hub_download(repo_id="dylanplummer/ropenet", filename="model.py", repo_type="model", token=os.environ['DATASET_SECRET'])
32
+ os.rename(model_file, "model.py")
33
  from model import RepNet
34
  # model_xml = hf_hub_download(repo_id="dylanplummer/ropenet", filename="model.xml", repo_type="model", token=os.environ['DATASET_SECRET'])
35
  # hf_hub_download(repo_id="dylanplummer/ropenet", filename="model.mapping", repo_type="model", token=os.environ['DATASET_SECRET'])
requirements.txt CHANGED
@@ -6,6 +6,7 @@ passlib
6
  scipy
7
  --find-links https://download.pytorch.org/whl/torch_stable.html
8
  opencv-python-headless==4.7.0.68
9
- openvino-dev==2022.3.0
10
- torch==1.13.1+cpu
11
- torchvision==0.14.1+cpu
 
 
6
  scipy
7
  --find-links https://download.pytorch.org/whl/torch_stable.html
8
  opencv-python-headless==4.7.0.68
9
+ # openvino-dev==2022.3.0
10
+ torch
11
+ torchvision
12
+ timm