sedeba19 commited on
Commit
7de16f6
·
1 Parent(s): ca0bfd2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def main():
34
 
35
  # Make prediction
36
  prediction = model.predict(input_data)
37
- wine_type = "White" if prediction[0] == 1 else "Red"
38
 
39
  # Display prediction
40
  st.write('Predicted Wine Type:', wine_type)
 
34
 
35
  # Make prediction
36
  prediction = model.predict(input_data)
37
+ wine_type = "**White Wine**" if prediction[0] == 1 else "**Red Wine**"
38
 
39
  # Display prediction
40
  st.write('Predicted Wine Type:', wine_type)