Update utils.py

utils.py

@@ -11,9 +11,19 @@ tqdm.pandas()
 def predict_popularity(features):
     predictions = [None] * 2
     predictions[0], predictions[1] = rf_model.predict([features]), model.predict([features])
+    addToCsvAndTrain(trainset)
     return predictions
 
 
+def addToCsvAndTrain(trainset):
+    trainset = [
+        [trainset[0],trainset[1],trainset[2],trainset[3],trainset[4],trainset[5],trainset[6],trainset[7],
+         trainset[8],trainset[9],trainset[10],trainset[11],trainset[12],trainset[13]
+         ]
+    ]
+    neues_df = pd.DataFrame(trainset, columns= data.columns)
+    df = pd.concat([data, neues_df], ignore_index=True)
+    df.to_csv('top50.csv', index=False)
 
 
 data = pd.read_csv('top50.csv', encoding='ISO-8859-1')