Make it possible to load skops model format

make-data.py

@@ -3,6 +3,7 @@
 import pickle
 
 import pandas as pd
+import skops.io as sio
 from sklearn.datasets import make_classification
 from sklearn.linear_model import LogisticRegression
 from sklearn.pipeline import Pipeline
@@ -21,6 +22,7 @@ clf.fit(X, y)
 
 with open("logreg.pkl", "wb") as f:
     pickle.dump(clf, f)
+sio.dump(clf, "logreg.skops")
 
 
 df.to_csv("data.csv", index=False)

start.py

@@ -45,7 +45,10 @@ def load_model() -> None:
         return
 
     bytes_data = st.session_state.model_file.getvalue()
-    model = pickle.loads(bytes_data)
+    if st.session_state.model_file.name.endswith("skops"):
+        model = sio.loads(bytes_data, trusted=True)
+    else:
+        model = pickle.loads(bytes_data)
     assert isinstance(model, BaseEstimator), "model must be an sklearn model"
 
     st.session_state.model = model
@@ -167,7 +170,11 @@ def start_input_form():
     )
 
     if not st.session_state.get("model_file"):
-        st.file_uploader("Upload a model*", on_change=load_model, key="model_file")
+        st.file_uploader(
+            "Upload an sklearn model (pickle or skops format)",
+            on_change=load_model,
+            key="model_file",
+        )
 
     st.markdown("---")
 
@@ -176,7 +183,7 @@ def start_input_form():
         "This sample data can be attached to the metadata of the model card"
     )
     st.file_uploader(
-        "Upload X data (csv)*", type=["csv"], on_change=load_data, key="data_file"
+        "Upload input data (csv)", type=["csv"], on_change=load_data, key="data_file"
     )
     st.markdown("---")