Update app.py

app.py

@@ -82,7 +82,32 @@ def theme_classification(text, text_classifier):
 sid_obj = SentimentIntensityAnalyzer()
 classifier = pipeline("zero-shot-classification", model="joeddav/xlm-roberta-large-xnli")
 df = load_dataset('csv', data_files=["Hugging Face DF.csv"], delimiter=',', column_names=["date", "url", "texts", "is_title", "theme", "translated"])
-#df = pd.DataFrame(dataset)
+df = pd.DataFrame(dataset)
+new_df = pd.DataFrame()
+
+dates = []
+is_title = []
+texts = []
+themes = []
+trans_text = []
+url = []
+
+for i in range(len(dataset)):
+    dates.append(dataset["date"][i])
+    is_title.append(dataset["is_title"][i])
+    texts.append(dataset["texts"][i])
+    themes.append(dataset["theme"][i])
+    trans_text.append(dataset["translated"][i])
+    url.append(dataset["url"][i])   
+
+new_df["date"] = dates
+new_df["text"] = texts
+new_df["theme"] = themes
+new_df["translated"] = trans_text
+new_df["is_title"] = is_title
+new_df["url"] = url
+
+df = new_df[:]
 
 header = st.container()
 model = st.container()