Create app.py

app.py

@@ -0,0 +1,45 @@
+import streamlit as st
+import numpy as np
+import pandas as pd
+from sklearn.linear_model import LinearRegression
+from sklearn.model_selection import train_test_split
+from sklearn.datasets import fetch_california_housing
+import pickle
+
+# Load the data
+california = fetch_california_housing()
+df = pd.DataFrame(california.data, columns=california.feature_names)
+df['MedHouseVal'] = california.target
+
+# Prepare the data for the model
+X = df[['MedInc']]
+y = df['MedHouseVal']
+
+# Split the data into training and testing sets
+X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
+
+# Train the model
+model = LinearRegression()
+model.fit(X_train, y_train)
+
+# Save the model
+with open("linear_regression_model.pkl", "wb") as file:
+    pickle.dump(model, file)
+
+# Load the model
+with open("linear_regression_model.pkl", "rb") as file:
+    model = pickle.load(file)
+
+# Streamlit app
+st.title('California Housing Price Prediction')
+
+med_inc = st.number_input('Enter Median Income:', min_value=0.0, step=0.01)
+
+if st.button('Predict'):
+    X_new = np.array([[med_inc]])
+    prediction = model.predict(X_new)
+    st.write(f'Predicted Median House Value: {prediction[0]}')
+
+# Display data
+if st.checkbox('Show raw data'):
+    st.write(df)