Add logs

app.py

@@ -1,6 +1,8 @@
 from fastapi import FastAPI
 from transformers import pipeline
 import os
+import logging
+
 
 # Set Hugging Face cache directory
 os.environ["HF_HOME"] = "/app/.cache"
@@ -14,6 +16,8 @@ async def root():
 
 @app.post("/predict")
 async def predict(data: dict):
+    logging.info(f"Received data: {data}")
     labels = ["Mathematics", "Language Arts", "Social Studies", "Science"]
     result = classifier(data["text"], labels)
+    logging.info(f"Prediction result: {result}")
     return {"label": result["labels"][0]}