Update app.py

app.py

@@ -68,6 +68,17 @@ avg_temp=st.slider('Slide The value to get Avg Temperature',min_value=1,max_valu
 model=joblib.load("model_rf.joblib")
 
 if st.button("Enter"):
-    output=model.predict([[Area_value,Item_value,avg_rainfall,pes_tonnes,avg_temp]])
+    try:
+        # Prepare input for the model
+        input_features = [[Area_value, Item_value, avg_rainfall, pes_tonnes, avg_temp]]
+        
+        # Get prediction from the model
+        output = model.predict(input_features)[0]
+
+        # Display the result with styling
+        st.success(f"🌱 The predicted **hg/ha_yield** is: **{output:.2f}**")
     
-    st.write("The value of hg/ha_yield:",output)
+    except Exception as e:
+        st.error(f"⚠️ Error in prediction: {e}")
+
+    st.write("")