Update app.py

app.py

@@ -1,5 +1,5 @@
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, TrainingArguments
-from datasets import load_dataset
+from datasets import pandas, Dataset
 import csv
 
 # Read requirements.txt file
@@ -19,8 +19,11 @@ with open('IMDB Dataset.csv', 'r') as csv_file:
         }
         preprocessed_data.append(preprocessed_entry)
 
-# Convert the preprocessed data to a dataset
-dataset = load_dataset('csv', data=preprocessed_data, delimiter=',')
+# Convert the preprocessed data to a pandas DataFrame
+df = pandas.DataFrame(preprocessed_data)
+
+# Convert the DataFrame to a datasets dataset
+dataset = Dataset.from_pandas(df)
 
 # Tokenize the dataset
 tokenizer = AutoTokenizer.from_pretrained("bigscience/bloom-560m")
@@ -38,3 +41,4 @@ import numpy as np
 import evaluate
 
 metric = evaluate.load("accuracy")
+