Spaces:
Sleeping
Sleeping
Update fastapi_app/app.py
Browse files- fastapi_app/app.py +1 -6
fastapi_app/app.py
CHANGED
|
@@ -17,12 +17,8 @@ from scripts.data_model import (
|
|
| 17 |
PoseClassificationResponse,
|
| 18 |
PosePrediction,
|
| 19 |
)
|
| 20 |
-
from scripts.s3 import download_model_from_s3
|
| 21 |
from scripts.huggingface_load import download_model_from_huggingface
|
| 22 |
|
| 23 |
-
# Toggle between S3 and Hugging Face model loading
|
| 24 |
-
# Set USE_HUGGINGFACE_MODELS = False to use S3 loader (production)
|
| 25 |
-
# Set USE_HUGGINGFACE_MODELS = True to use Hugging Face loader (Spaces deployment)
|
| 26 |
USE_HUGGINGFACE_MODELS = True
|
| 27 |
|
| 28 |
warnings.filterwarnings("ignore")
|
|
@@ -70,8 +66,7 @@ def initialize_model():
|
|
| 70 |
logger.info(f"Downloading model from Hugging Face to {LOCAL_MODEL_PATH}")
|
| 71 |
success = download_model_from_huggingface(LOCAL_MODEL_PATH)
|
| 72 |
else:
|
| 73 |
-
logger.info(
|
| 74 |
-
success = download_model_from_s3(LOCAL_MODEL_PATH, f"{MODEL_NAME}/")
|
| 75 |
|
| 76 |
if not success:
|
| 77 |
logger.error("Failed to download model")
|
|
|
|
| 17 |
PoseClassificationResponse,
|
| 18 |
PosePrediction,
|
| 19 |
)
|
|
|
|
| 20 |
from scripts.huggingface_load import download_model_from_huggingface
|
| 21 |
|
|
|
|
|
|
|
|
|
|
| 22 |
USE_HUGGINGFACE_MODELS = True
|
| 23 |
|
| 24 |
warnings.filterwarnings("ignore")
|
|
|
|
| 66 |
logger.info(f"Downloading model from Hugging Face to {LOCAL_MODEL_PATH}")
|
| 67 |
success = download_model_from_huggingface(LOCAL_MODEL_PATH)
|
| 68 |
else:
|
| 69 |
+
logger.info("failed to download model")
|
|
|
|
| 70 |
|
| 71 |
if not success:
|
| 72 |
logger.error("Failed to download model")
|