Update app.py

app.py

@@ -5,6 +5,7 @@ import pickle
 import torch
 import time
 import pandas as pd
+import sklearn 
 
 # Load ML model
 model = pickle.load(open('model.pkl', 'rb'))