frist commit

app.py

@@ -22,16 +22,15 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-
-MODEL_PATH = "ml-models/"  
-sentiment_model_path = "tinybert-sentiment-analysis/"
-disaster_model_path = "tinybert-disaster-tweet/"
-image_model_path = "vit-human-pose-classification/"
+ 
+sentiment_model_path = "ml-models/tinybert-sentiment-analysis/"
+disaster_model_path = "ml-models/tinybert-disaster-tweet/"
+image_model_path = "ml-models/vit-human-pose-classification/"
 
 logger.info("Ensuring models are downloaded...")
-download_model_from_s3(MODEL_PATH + sentiment_model_path, sentiment_model_path)
-download_model_from_s3(MODEL_PATH + disaster_model_path, disaster_model_path)
-download_model_from_s3(MODEL_PATH + image_model_path, image_model_path)
+download_model_from_s3(sentiment_model_path, sentiment_model_path)
+download_model_from_s3(disaster_model_path, disaster_model_path)
+download_model_from_s3(image_model_path, image_model_path)
 logger.info("All models are ready.")
 
 

scripts/s3.py

@@ -1,11 +1,12 @@
 import os
 import boto3
-from pathlib import Path
 from utils.log import logger
 
 aws_access_key = os.getenv("AWS_ACCESS_KEY_ID") 
 aws_key_pw =  os.getenv("AWS_SECRET_ACCESS_KEY") 
 BUCKET_NAME =  os.getenv("BUCKET_NAME") 
+
+
 s3 = boto3.client(
     "s3",
     aws_access_key_id=aws_access_key,
@@ -13,7 +14,7 @@ s3 = boto3.client(
 )
 
 
-def download_model_from_s3(local_path: Path, s3_prefix: str):
+def download_model_from_s3(local_path: str, s3_prefix: str):
     try:
         if os.path.exists(local_path) and os.listdir(local_path):
             logger.info(f"Model {local_path} already exists. Skipping download.")
@@ -30,7 +31,7 @@ def download_model_from_s3(local_path: Path, s3_prefix: str):
                     local_file = os.path.join(local_path, os.path.relpath(s3_key, s3_prefix))
                    
                     os.makedirs(os.path.dirname(local_file), exist_ok=True)
-                    logger.info(f"Downloading {s3_key} to {local_file}")
+                    logger.info(f"Downloading model from {BUCKET_NAME} key {s3_key} to {local_file}")
                     s3.download_file(BUCKET_NAME, s3_key, local_file)
                     logger.info(f"Completed download {s3_key} to {local_file}")
     except Exception as e: