Spaces:

fahmiaziz
/

ml-model-restapi

Sleeping

fahmiaziz98 commited on Apr 3

Commit

57d9736

1 Parent(s): 583e7cf

frist commit

Files changed (3) hide show

router/disaster.py CHANGED Viewed

@@ -8,7 +8,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/")
 @router.post(
     "/disaster_classification",

 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")
 @router.post(
     "/disaster_classification",

router/image_clf.py CHANGED Viewed

@@ -8,7 +8,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/")
 @router.post(
     "/image_classification",

 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")
 @router.post(
     "/image_classification",

router/sentiment.py CHANGED Viewed

@@ -8,7 +8,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/")
 @router.post(
     "/sentiment_classification",

 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")
 @router.post(
     "/sentiment_classification",