set model_repo to use env variable

app.py

@@ -5,9 +5,10 @@ from predict_dept_model import DepartmentPredictor
 from contextlib import asynccontextmanager
 from response_schema import ClassificationOutput, TextInput 
 from huggingface_hub import  HfApi
+import os
 
 # Define the model repository ID
-model_repo = "mr-kush/sambodhan-department-classification-model"
+model_repo = os.getenv("MODEL_REPO")
 
 # hf api
 api = HfApi()
@@ -19,7 +20,7 @@ api = HfApi()
 async def lifespan(app: FastAPI):
     # Load the model 
     global predictor
-    predictor = DepartmentPredictor()
+    predictor = DepartmentPredictor(model_repo= model_repo)
     yield