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()
|