Spaces:

Add1E
/

SpotifyHitPrediction

Sleeping

Add1E commited on Nov 15, 2023

Commit

738d8c0

1 Parent(s): 1971b23

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -8,7 +8,7 @@ import streamlit as st
 tqdm.pandas()
-def predict_popularity(features):
     predictions = [None] * 2
     predictions[0], predictions[1] = rf_model.predict([features]), model.predict([features])
     addToCsvAndTrain(trainset)

 tqdm.pandas()
+def predict_popularity(features, trainset):
     predictions = [None] * 2
     predictions[0], predictions[1] = rf_model.predict([features]), model.predict([features])
     addToCsvAndTrain(trainset)