Update app.py

app.py

@@ -2,6 +2,7 @@
 import streamlit as st
 import mlflow
 import pandas as pd
+from mlflow.tracking import MlflowClient
 
 # Set the MLflow tracking URI (e.g., local file system or a remote server)
 mlflow.set_tracking_uri("http://localhost:5000")
@@ -9,8 +10,11 @@ mlflow.set_tracking_uri("http://localhost:5000")
 def main():
     st.title("Streamlit and MLflow Integration")
 
+    # Create an MlflowClient instance
+    client = MlflowClient()
+
     # Retrieve the list of experiments
-    experiments = mlflow.list_experiments()
+    experiments = client.list_experiments()
     experiment_names = [exp.name for exp in experiments]
     experiment_id_map = {exp.name: exp.experiment_id for exp in experiments}