Spaces:
Sleeping
Sleeping
Update image_captioning.py
Browse files- image_captioning.py +1 -1
image_captioning.py
CHANGED
|
@@ -29,7 +29,7 @@ if DEVICE == "cuda":
|
|
| 29 |
|
| 30 |
# Model configuration
|
| 31 |
MODEL_NAME = 'iic/mPLUG-Owl3-7B-240728'
|
| 32 |
-
MODEL_CACHE_DIR = "data/models"
|
| 33 |
os.makedirs(MODEL_CACHE_DIR, exist_ok=True)
|
| 34 |
|
| 35 |
# Download and cache the model
|
|
|
|
| 29 |
|
| 30 |
# Model configuration
|
| 31 |
MODEL_NAME = 'iic/mPLUG-Owl3-7B-240728'
|
| 32 |
+
MODEL_CACHE_DIR = "/data/models"
|
| 33 |
os.makedirs(MODEL_CACHE_DIR, exist_ok=True)
|
| 34 |
|
| 35 |
# Download and cache the model
|