Update app.py

app.py

@@ -10,6 +10,8 @@ from transformers import (
 from datasets import load_dataset, Dataset
 import torch
 import pandas as pd
+from sklearn.model_selection import train_test_split
+
 
 # Configuration
 MODEL_NAME = "microsoft/DialoGPT-medium"