ferdmartin commited on
Commit
997f16f
·
verified ·
1 Parent(s): 2997dbb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -62,11 +62,11 @@ def predict_rating(reviewerID, itemID, model):
62
  model = load_model()
63
 
64
  # Streamlit app interface
65
- st.title('Product Rating Prediction')
66
 
67
  # Example input values
68
- # example_reviewerID = "10" # Example reviewerID
69
- # example_itemID = "0" # Example itemID
70
 
71
  # User inputs with examples
72
  # reviewerID = st.text_input('Reviewer ID', value=example_reviewerID)
@@ -77,4 +77,4 @@ itemID = st.text_input('Item ID')
77
  # Predict button
78
  if st.button('Predict Rating'):
79
  prediction = predict_rating(reviewerID, itemID, model)
80
- st.write(f'Predicted Rating: {prediction:.2f}')
 
62
  model = load_model()
63
 
64
  # Streamlit app interface
65
+ st.title('Music Rating Prediction - Amazon Review')
66
 
67
  # Example input values
68
+ example_reviewerID = "61658" # Example reviewerID
69
+ example_itemID = "5000" # Example itemID
70
 
71
  # User inputs with examples
72
  # reviewerID = st.text_input('Reviewer ID', value=example_reviewerID)
 
77
  # Predict button
78
  if st.button('Predict Rating'):
79
  prediction = predict_rating(reviewerID, itemID, model)
80
+ st.write(f'Predicted Rating: {prediction:.2f}')