Update for predict

app.py

@@ -1,13 +1,17 @@
-import os
 from fastapi import FastAPI
 from transformers import pipeline
+import os
 
-# Set a custom Hugging Face cache directory
-os.environ["TRANSFORMERS_CACHE"] = "/app/.cache"
+# Set Hugging Face cache directory
+os.environ["HF_HOME"] = "/app/.cache"
 
 app = FastAPI()
 classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
 
+@app.get("/")
+async def read_root():
+    return {"message": "Welcome to the Zero-Shot Classification API"}
+
 @app.post("/predict")
 async def predict(data: dict):
     labels = ["Mathematics", "Language Arts", "Social Studies", "Science"]