Update app.py

app.py

@@ -8,6 +8,7 @@ from sklearn import preprocessing
 from sklearn.model_selection import train_test_split
 from sklearn.ensemble import RandomForestClassifier
 from sklearn.metrics import accuracy_score
+import os
 
 HF_TOKEN = os.getenv("titanic_dataset")
 hf_writer = gr.HuggingFaceDatasetSaver(hf_foken=HF_TOKEN, dataset_name="AllAideas/titanic_dataset")