fix
Browse files
main.py
CHANGED
|
@@ -41,7 +41,7 @@ app.add_middleware(
|
|
| 41 |
ANIMALS = ['Cat', 'Dog', 'Panda'] # Animal names here, these represent the labels of the images that we trained our model on.
|
| 42 |
|
| 43 |
# 1) download your SavedModel from the Hub into a writable directory (Spaces often
|
| 44 |
-
#
|
| 45 |
repo_id = "IDS75912/masterclass-2025"
|
| 46 |
local_model_dir = os.environ.get('HF_MODEL_DIR', './model_dir')
|
| 47 |
|
|
|
|
| 41 |
ANIMALS = ['Cat', 'Dog', 'Panda'] # Animal names here, these represent the labels of the images that we trained our model on.
|
| 42 |
|
| 43 |
# 1) download your SavedModel from the Hub into a writable directory (Spaces often
|
| 44 |
+
# take HF_MODEL_DIR or default model_dir).
|
| 45 |
repo_id = "IDS75912/masterclass-2025"
|
| 46 |
local_model_dir = os.environ.get('HF_MODEL_DIR', './model_dir')
|
| 47 |
|