Update inference_server.py

inference_server.py

@@ -1,3 +1,4 @@
+import logging 
 from sklearn.linear_model import SGDClassifier
 import uvicorn
 from fastapi import FastAPI
@@ -20,4 +21,11 @@ def train(X):
 
 @app.get("/")
 def read_root(input_text):
-    return predict(input_text)
+    logging.info("Received request with input_text: %s"), input_text)
+    try:
+        result = predict(input_text)
+        logging.info("Prediction made: %s", result)
+        return result
+    except Exception as e:
+        logging.error("An error occured: %s", e)
+        return {"error": str(e)}