Update app.py

app.py

@@ -10,6 +10,13 @@ from sklearn.model_selection import train_test_split
 from sklearn.metrics import classification_report, accuracy_score
 import gradio as gr
 
+# Load dataset
+file_path = 'spam_ham_dataset.csv'
+df = pd.read_csv(file_path)
+
+# Convert label column to numeric (0 for ham, 1 for spam)
+df['label_num'] = df['label'].astype('category').cat.codes
+
 # Define device
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")