Update app.py

app.py

@@ -7,8 +7,20 @@ app = FastAPI()
 
 token = os.environ.get("HF_TOKEN")
 
-english_model = pipeline("text-classification", model="mrgmd01/Finetuned_siebert-sentiment-roberta-large-english", token=token)
-urdu_model = pipeline("text-classification", model="mrgmd01/SA_Model_bert-base-multilingual-uncased", token=token)
+english_model = None
+urdu_model = None
+
+def get_english_model():
+    global english_model
+    if english_model is None:
+        english_model = pipeline("text-classification", model="mrgmd01/Finetuned_siebert-sentiment-roberta-large-english", token=token)
+    return english_model
+
+def get_urdu_model():
+    global urdu_model
+    if urdu_model is None:
+        urdu_model = pipeline("text-classification", model="mrgmd01/SA_Model_bert-base-multilingual-uncased", token=token)
+    return urdu_model
 
 @app.post("/predict")
 async def predict(request: Request):
@@ -17,8 +29,8 @@ async def predict(request: Request):
     language = body.get("language", "english")
     
     if language in ["urdu", "roman_urdu"]:
-        result = urdu_model(text)[0]
+        result = get_urdu_model()(text)[0]
     else:
-        result = english_model(text)[0]
+        result = get_english_model()(text)[0]
     
     return JSONResponse({"label": result["label"], "score": result["score"]})