Change label data types from str to int

app.py

@@ -11,7 +11,10 @@ sample_data = [
     ("Attended a seminar on non-surgical treatments.", "non-surgery"),
 ]
 
+# Map labels to integers
+label_mapping = {"surgery": 1, "non-surgery": 0}
 train_texts, train_labels = zip(*sample_data)
+train_labels = [label_mapping[label] for label in train_labels]
 
 # Logging and Outputs
 st.write("Sample data:")