Spaces:
Runtime error
Runtime error
Update caption_store.py
Browse files- caption_store.py +4 -1
caption_store.py
CHANGED
|
@@ -5,7 +5,10 @@ import os
|
|
| 5 |
import time
|
| 6 |
from typing import Optional
|
| 7 |
|
| 8 |
-
|
|
|
|
|
|
|
|
|
|
| 9 |
|
| 10 |
|
| 11 |
def _load() -> dict:
|
|
|
|
| 5 |
import time
|
| 6 |
from typing import Optional
|
| 7 |
|
| 8 |
+
if os.path.exists("/data"):
|
| 9 |
+
CAPTION_STORE_PATH = "/data/captions.json"
|
| 10 |
+
else:
|
| 11 |
+
CAPTION_STORE_PATH = "captions.json"
|
| 12 |
|
| 13 |
|
| 14 |
def _load() -> dict:
|