Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ st.title("Metric Compare")
|
|
| 13 |
st.markdown("### Choose the dataset you want to use for the comparison:")
|
| 14 |
|
| 15 |
api = HfApi()
|
| 16 |
-
dsets = [d.id for d in api.list_datasets(filter="task_categories:text-classification", sort = "downloads", direction=-1, limit = 20)]
|
| 17 |
|
| 18 |
dset = st.selectbox('Choose a dataset from the Hub', options=dsets)
|
| 19 |
|
|
@@ -37,29 +37,15 @@ models = st.multiselect(
|
|
| 37 |
'Choose the models that have been trained/finetuned on this dataset',
|
| 38 |
options=all_models)
|
| 39 |
|
| 40 |
-
button = st.button("Print Models",disabled=False)
|
| 41 |
|
| 42 |
-
if button :
|
| 43 |
-
if len(location) < 6:
|
| 44 |
-
st.write(models)
|
| 45 |
-
else:
|
| 46 |
-
st.warning("Please select at most 5 models")
|
| 47 |
-
|
| 48 |
st.markdown("### What two metrics do you want to compare?")
|
| 49 |
|
| 50 |
metrics = st.multiselect(
|
| 51 |
'Choose the metrics for the comparison',
|
| 52 |
options=['f1', 'accuracy', 'precision', 'recall'])
|
| 53 |
|
| 54 |
-
button2 = st.button("Print Metrics",disabled=False)
|
| 55 |
-
|
| 56 |
-
if button2 :
|
| 57 |
-
if len(metrics ) < 3:
|
| 58 |
-
st.write(metrics)
|
| 59 |
-
else:
|
| 60 |
-
st.warning("Please select at most 2 metrics")
|
| 61 |
|
| 62 |
-
st.markdown("###
|
| 63 |
|
| 64 |
### Loading data
|
| 65 |
try:
|
|
|
|
| 13 |
st.markdown("### Choose the dataset you want to use for the comparison:")
|
| 14 |
|
| 15 |
api = HfApi()
|
| 16 |
+
dsets = [d.id for d in api.list_datasets(filter="task_categories:text-classification", sort = "downloads", direction=-1, limit = 20) if d.id not 'glue']
|
| 17 |
|
| 18 |
dset = st.selectbox('Choose a dataset from the Hub', options=dsets)
|
| 19 |
|
|
|
|
| 37 |
'Choose the models that have been trained/finetuned on this dataset',
|
| 38 |
options=all_models)
|
| 39 |
|
|
|
|
| 40 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
st.markdown("### What two metrics do you want to compare?")
|
| 42 |
|
| 43 |
metrics = st.multiselect(
|
| 44 |
'Choose the metrics for the comparison',
|
| 45 |
options=['f1', 'accuracy', 'precision', 'recall'])
|
| 46 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 47 |
|
| 48 |
+
st.markdown("### Please wait for the dataset and models to load (this can take some time if they are big!")
|
| 49 |
|
| 50 |
### Loading data
|
| 51 |
try:
|