ml model

app.py

@@ -13,10 +13,7 @@ def load_pickle(filename):
         return data
     
 # Load pickle file
-ml_components = load_pickle('my_model.pkl') 
-
-# Components in the pickle file
-ml_model = ml_components['model']
+ml_model = load_pickle('my_model.pkl') 
 
 #Endpoints 
 #Root endpoints