AlignAI commited on
Commit
e059a6a
·
verified ·
1 Parent(s): 57048f9

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +49 -0
app.py ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import joblib
3
+ import numpy as np
4
+
5
+ # Load the trained model and scaler
6
+ # Use @st.cache_resource to load them only once for performance
7
+ @st.cache_resource
8
+ def load_model():
9
+ model = joblib.load('svm_model.pkl')
10
+ scaler = joblib.load('scaler.pkl')
11
+ return model, scaler
12
+
13
+ try:
14
+ model, scaler = load_model()
15
+ except FileNotFoundError:
16
+ st.error("Model files not found. Please run train_model.py first.")
17
+ st.stop()
18
+
19
+ st.title("Purchase Intention Predictor")
20
+ st.write("Adjust the sliders below to predict the user's Purchase Intention (PI).")
21
+
22
+ # Sidebar for inputs
23
+ st.sidebar.header("User Inputs")
24
+
25
+ # Create sliders for each feature based on the data's 1-7 scale
26
+ att = st.sidebar.slider("Attitude (ATT)", min_value=1.0, max_value=7.0, value=4.0, step=0.1)
27
+ sns = st.sidebar.slider("Subjective Norms (SNs)", min_value=1.0, max_value=7.0, value=4.0, step=0.1)
28
+ pbc = st.sidebar.slider("Perceived Behavioral Control (PBC)", min_value=1.0, max_value=7.0, value=4.0, step=0.1)
29
+ eo = st.sidebar.slider("Environmental Outcome (EO)", min_value=1.0, max_value=7.0, value=4.0, step=0.1)
30
+ ec = st.sidebar.slider("Environmental Concern (EC)", min_value=1.0, max_value=7.0, value=4.0, step=0.1)
31
+
32
+ # Prepare input data
33
+ input_data = np.array([[att, sns, pbc, eo, ec]])
34
+
35
+ # Scale the input
36
+ input_scaled = scaler.transform(input_data)
37
+
38
+ # Predict
39
+ if st.button("Predict Purchase Intention"):
40
+ prediction = model.predict(input_scaled)
41
+ st.subheader(f"Predicted Purchase Intention Score: {prediction[0]:.2f}")
42
+
43
+ # Optional: Interpretation
44
+ if prediction[0] > 5.5:
45
+ st.success("High Purchase Intention")
46
+ elif prediction[0] < 3.5:
47
+ st.warning("Low Purchase Intention")
48
+ else:
49
+ st.info("Moderate Purchase Intention")