Update app.py

app.py

@@ -39,7 +39,7 @@ except Exception as e:
     validation_data = pd.DataFrame()
 
 # Ensure the necessary columns exist in the training dataset
-required_columns = ["text", "Ground_Truth_Stress", "Ground_Truth_Anxiety", "Ground_Truth_Depression", "Ground_Truth_Other_binary"]
+required_columns = ["content", "Ground_Truth_Stress", "Ground_Truth_Anxiety", "Ground_Truth_Depression", "Ground_Truth_Other_binary"]
 if not train_data.empty:
     for column in required_columns:
         if column not in train_data.columns:
@@ -49,7 +49,7 @@ else:
     exit(1)
 
 # Initialize BM25
-tokenized_train = [doc.split() for doc in train_data["text"]]
+tokenized_train = [doc.split() for doc in train_data["content"]]
 bm25 = BM25Okapi(tokenized_train)
 
 # Set OpenAI API key