Spaces:
Running
Running
nicpopovic
commited on
Commit
•
d1e7ead
1
Parent(s):
9947d5c
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,6 @@ import numpy as np
|
|
9 |
import os
|
10 |
import urllib.request
|
11 |
import zipfile
|
12 |
-
import collections
|
13 |
|
14 |
|
15 |
class MLP(torch.nn.Module):
|
@@ -146,7 +145,7 @@ def find_datasets_and_model_ids(root_dir):
|
|
146 |
dataset_name = os.path.basename(os.path.dirname(stoke_config_path))
|
147 |
datasets.setdefault(model_id, {})[dataset_name] = stoke_config_data
|
148 |
|
149 |
-
return
|
150 |
|
151 |
|
152 |
# Main content
|
@@ -166,7 +165,7 @@ with st.sidebar:
|
|
166 |
# Select dataset based on selected model
|
167 |
available_datasets = datasets[model_selection]
|
168 |
if available_datasets:
|
169 |
-
dataset_selection = st.selectbox("Select Dataset", available_datasets)
|
170 |
else:
|
171 |
st.error("No datasets available for the selected model.")
|
172 |
|
|
|
9 |
import os
|
10 |
import urllib.request
|
11 |
import zipfile
|
|
|
12 |
|
13 |
|
14 |
class MLP(torch.nn.Module):
|
|
|
145 |
dataset_name = os.path.basename(os.path.dirname(stoke_config_path))
|
146 |
datasets.setdefault(model_id, {})[dataset_name] = stoke_config_data
|
147 |
|
148 |
+
return datasets
|
149 |
|
150 |
|
151 |
# Main content
|
|
|
165 |
# Select dataset based on selected model
|
166 |
available_datasets = datasets[model_selection]
|
167 |
if available_datasets:
|
168 |
+
dataset_selection = st.selectbox("Select Dataset", sorted(available_datasets))
|
169 |
else:
|
170 |
st.error("No datasets available for the selected model.")
|
171 |
|