Update app.py

app.py

@@ -484,7 +484,7 @@ def main():
   #     fi.write(dat)
   #     fi.truncate()
 
-  #updated_dataset = dataset.map(lambda example: {'new_value': level2_df['value'], 'new_source':level2_df['source'], 'new_target': level2_df['target']},remove_columns=['value','source','target'])
+  updated_dataset = dataset.map(lambda example: {'new_value': level2_df['value'], 'new_source':level2_df['source'], 'new_target': level2_df['target']},remove_columns=['value','source','target'])
 
   df_final1.to_csv('predictions.csv')
   csv_file = "predictions.csv"