codeby-hp commited on
Commit
c006e51
·
verified ·
1 Parent(s): fc54201

Update fastapi_app/app.py

Browse files
Files changed (1) hide show
  1. 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(f"Downloading model from S3 to {LOCAL_MODEL_PATH}")
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")