Update app.py

app.py

@@ -2,7 +2,10 @@ import gradio as gr
 from fastapi import FastAPI
 from risk_model import predict_risk, retrain_model, get_history_df
 
+# FastAPI backend
 app = FastAPI()
+
+# Gradio UI
 gradio_app = gr.Blocks()
 
 with gradio_app:
@@ -32,9 +35,9 @@ with gradio_app:
     predict_btn.click(classify, inputs=[temp, duration], outputs=[result, score, history_table])
     retrain_btn.click(retrain_model, outputs=[retrain_output])
 
-# Mount Gradio onto FastAPI
+# Mount Gradio UI on FastAPI
 @app.get("/")
-def read_root():
-    return {"message": "Heating Mantle Risk API is running!"}
+def root():
+    return {"message": "Heating Mantle Risk API is live!"}
 
 app = gr.mount_gradio_app(app, gradio_app, path="/predict-ui")