updating the Dockerfile

main.py

@@ -74,4 +74,8 @@ def predict_endpoint(input_data: InputData):
 
     return {
         "prediction": float(result)
-    }
+    }
+
+@app.get("/")
+def health_check():
+    return {"healthy": True}