dylanplummer commited on
Commit
f142dbc
·
1 Parent(s): 50019e0

make nonlocal

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -30,13 +30,13 @@ from hls_download import download_clips
30
 
31
  #plt.style.use('dark_background')
32
 
33
- LOCAL = True
34
  IMG_SIZE = 256
35
  CACHE_API_CALLS = False
36
  os.makedirs(os.path.join(os.getcwd(), 'clips'), exist_ok=True)
37
  current_model = 'nextjump_speed'
38
- #onnx_file = hf_hub_download(repo_id="lumos-motion/nextjump", filename=f"{current_model}.onnx", repo_type="model", token=os.environ['DATASET_SECRET'])
39
- onnx_file = f'{current_model}.onnx'
40
  api = HfApi()
41
 
42
  if torch.cuda.is_available():
 
30
 
31
  #plt.style.use('dark_background')
32
 
33
+ LOCAL = False
34
  IMG_SIZE = 256
35
  CACHE_API_CALLS = False
36
  os.makedirs(os.path.join(os.getcwd(), 'clips'), exist_ok=True)
37
  current_model = 'nextjump_speed'
38
+ onnx_file = hf_hub_download(repo_id="lumos-motion/nextjump", filename=f"{current_model}.onnx", repo_type="model", token=os.environ['DATASET_SECRET'])
39
+ #onnx_file = f'{current_model}.onnx'
40
  api = HfApi()
41
 
42
  if torch.cuda.is_available():