frist commit

router/disaster.py

@@ -7,8 +7,7 @@ from utils.pipeline import load_model
 
 router = APIRouter()
 
-BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))  
-MODEL_PATH = os.path.join(BASE_DIR, "ml_models", "tinybert-disaster-tweet")
+MODEL_PATH = "ml_models/tinybert-disaster-tweet/"
 
 @router.post(
     "/disaster_classification",

router/image_clf.py

@@ -7,8 +7,7 @@ from utils.pipeline import load_model
 
 router = APIRouter()
 
-BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))  
-MODEL_PATH = os.path.join(BASE_DIR, "ml_models", "vit-human-pose-classification")
+MODEL_PATH = "ml_models/vit-human-pose-classification/"
 
 @router.post(
     "/image_classification",

router/sentiment.py

@@ -7,8 +7,7 @@ from utils.pipeline import load_model
 
 router = APIRouter()
 
-BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))  
-MODEL_PATH = os.path.join(BASE_DIR, "ml_models", "tinybert-sentiment-analysis")
+MODEL_PATH = "ml_models/tinybert-sentiment-analysis/"
 
 @router.post(
     "/sentiment_classification",