Commit ·
f142dbc
1
Parent(s): 50019e0
make nonlocal
Browse files
app.py
CHANGED
|
@@ -30,13 +30,13 @@ from hls_download import download_clips
|
|
| 30 |
|
| 31 |
#plt.style.use('dark_background')
|
| 32 |
|
| 33 |
-
LOCAL =
|
| 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 |
-
|
| 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():
|