Update app.py

app.py

@@ -19,7 +19,8 @@ dset = st.selectbox('Choose a dataset from the Hub', options=datasets)
 
 st.markdown("## Now select up to 5 models to compare their performance:")
 
-models = api.list_models(filter="trained_dataset:imdb", sort = "downloads", direction=-1, limit = 20)
+filt = ModelFilter(trained_dataset=dset)
+models = api.list_models(filter=filt, sort = "downloads", direction=-1, limit = 20)
 
 model = st.multiselect(
      'Choose the models that have been trained/finetuned on this dataset',