Update app.py
Browse files
app.py
CHANGED
|
@@ -30,11 +30,11 @@ from hls_download import download_clips
|
|
| 30 |
#plt.style.use('dark_background')
|
| 31 |
|
| 32 |
LOCAL = False
|
| 33 |
-
IMG_SIZE =
|
| 34 |
CACHE_API_CALLS = False
|
| 35 |
os.makedirs(os.path.join(os.getcwd(), 'clips'), exist_ok=True)
|
| 36 |
|
| 37 |
-
onnx_file = hf_hub_download(repo_id="lumos-motion/nextjump", filename="
|
| 38 |
|
| 39 |
#onnx_file = 'nextjump.onnx'
|
| 40 |
|
|
|
|
| 30 |
#plt.style.use('dark_background')
|
| 31 |
|
| 32 |
LOCAL = False
|
| 33 |
+
IMG_SIZE = 256
|
| 34 |
CACHE_API_CALLS = False
|
| 35 |
os.makedirs(os.path.join(os.getcwd(), 'clips'), exist_ok=True)
|
| 36 |
|
| 37 |
+
onnx_file = hf_hub_download(repo_id="lumos-motion/nextjump", filename="nextjump_256.onnx", repo_type="model", token=os.environ['DATASET_SECRET'])
|
| 38 |
|
| 39 |
#onnx_file = 'nextjump.onnx'
|
| 40 |
|