Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -12,8 +12,8 @@ def main():
|
|
12 |
if clicked:
|
13 |
result=model.predict(pd.DataFrame({"age": [age],'Estimated Salary':[salary]}))
|
14 |
# Show prediction
|
15 |
-
result ='
|
16 |
-
st.success("You will" +result+" this product")
|
17 |
# Run main()
|
18 |
if __name__=="__main__":
|
19 |
main()
|
|
|
12 |
if clicked:
|
13 |
result=model.predict(pd.DataFrame({"age": [age],'Estimated Salary':[salary]}))
|
14 |
# Show prediction
|
15 |
+
result ='buy' if result[0]==1 else 'not buy'
|
16 |
+
st.success("You will " +result+" this product")
|
17 |
# Run main()
|
18 |
if __name__=="__main__":
|
19 |
main()
|