Update app.py

app.py

@@ -9,6 +9,9 @@ from sklearn.model_selection import train_test_split
 from sklearn.ensemble import RandomForestClassifier
 from sklearn.metrics import accuracy_score
 
+HF_TOKEN = os.getenv("titanic_dataset")
+hf_writer = gr.HuggingFaceDatasetSaver(hf_foken=HF_TOKEN, dataset_name="AllAideas/titanic_dataset")
+
 data = pd.read_csv('https://raw.githubusercontent.com/gradio-app/titanic/master/train.csv')
 data.head()
 
@@ -64,4 +67,6 @@ age = gr.inputs.Slider(minimum=0, maximum=120, default=22, label="Edad")
 fare = gr.inputs.Slider(minimum=0, maximum=200, default=100, label="Clase")
 
 gr.Interface(predict_survival, [sex, age, fare], "label", live=True, thumbnail="https://raw.githubusercontent.com/gradio-app/hub-titanic/master/thumbnail.png", analytics_enabled=False,
-    title="Sobrevivientes del Titanic", description="Analicemos los sobreviventes de este caso y comprobemos").launch();
+    title="Sobrevivientes del Titanic", description="Analicemos los sobreviventes de este caso y comprobemos", allow_flagging="manual",
+					flagging_options=["Correcto", "Incorrecto", "Casi correcto", "Error", "Otro"],
+					flagging_callback=hf_writer).launch();