Update app.py

app.py

@@ -7,7 +7,7 @@ import pickle
 with open("model.pkl", "rb") as f:
     model = pickle.load(f)
 
-st.write(""" # Mobile Price-Range Prediction""")
+st.write(""" # Mobile Price-Range Prediction\n ## Prices ranges from 0 to 3""")
 st.sidebar.header("Choose Phone Specs")
 
 def input_features():
@@ -30,4 +30,4 @@ st.write(feats)
 if st.button("Result"):
     #res = requests.post(url = "http://0.0.0.0:8008/predict", data=json.dumps(data)).text
     res = model.predict(feats)
-    st.subheader(f"Predicted Price Range: {res}")
+    st.subheader(f"Predicted Price Range: {res[0]}")