Update app.py

app.py

@@ -15,7 +15,7 @@ def predict_price(location, sqft, bath, bhk):
     if loc_index >= 0:
         x[loc_index] = 1
 
-    return abs(lr_clf.predict([x])[0])
+    return round(abs(lr_clf.predict([x])[0]), 2)
 
 with open('./model.pkl', 'rb') as f:
     lr_clf = pickle.load(f)
@@ -33,7 +33,7 @@ def main():
     if st.button("Predict"):
         # Perform prediction
         prediction = predict_price(location, sqft, bath, bhk)
-        st.success(f"The predicted price is: {prediction:0.2f} lakhs")
+        st.success(f"The predicted price is: {prediction} lakhs")
 
 if __name__ == "__main__":
     main()