Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import joblib
 import pandas as pd
+from sklearn.model_selection import train_test_split
 from sklearn.metrics import r2_score, mean_absolute_error, mean_squared_error
 from typing import List, Dict, Any
 import gradio as gr