Update app.py

app.py

@@ -1,10 +1,8 @@
 import gradio as gr
-from datasets import load_dataset
+from datasets import load_dataset, Dataset
 import pandas as pd
 from huggingface_hub import create_repo
 from huggingface_hub import login
-import os 
-login(token='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')
 
 dataset = load_dataset("torileatherman/sentiment_analysis_batch_predictions", split='train')
 predictions_df = pd.DataFrame(dataset)
@@ -44,13 +42,16 @@ def manual_label():
     random_prediction = random_sample['Prediction'].iloc[0]
     return random_headline, random_prediction
 
+
+
 def thanks(sentiment):
     labeled_sentiments = []
     labeled_sentiments.append(sentiment)
     counter = len(labeled_sentiments)
-    # create repo
+    counter = str(counter)
+    login(token = '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')
     create_repo("torileatherman/"+counter+"labeled_data")
-    # push to repo
+    labeled_sentiments = pd.DataFrame(labeled_sentiments, columns = ['Predictions'])
     labeled_sentiments = Dataset.from_pandas(labeled_sentiments)
     labeled_sentiments.push_to_hub("torileatherman/"+counter+"labeled_data")
     return f"""Thank you for making our model better!"""