Update src/streamlit_app.py

src/streamlit_app.py

@@ -4,6 +4,7 @@ from sklearn.model_selection import train_test_split
 from sklearn.preprocessing import StandardScaler
 from sklearn.ensemble import RandomForestClassifier
 from sklearn.metrics import accuracy_score
+from datasets import load_dataset
 
 
 ds = load_dataset("codesignal/wine-quality")