nicpopovic commited on
Commit
34c1316
·
verified ·
1 Parent(s): 3a3ab25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -123,7 +123,7 @@ def find_datasets_and_model_ids(root_dir):
123
  if not os.path.exists(root_dir):
124
  # If root directory doesn't exist, download a zip file and unpack it
125
  print("Root directory doesn't exist. Downloading zip file...")
126
- url = "https://drive.usercontent.google.com/download?id=1i9PM1M3uSTiy5Ur43cGsK-KflhQVjg1Y&export=download&authuser=0&confirm=t&uuid=a8eda6d1-dc72-4c5c-a09a-44d3c8ef2ba7&at=APZUnTU9Mlam5kzHsQCwvNFfqF4H%3A1711095311813" # Replace with your actual download URL
127
  download_and_extract_zip(url, root_dir)
128
  print("Zip file downloaded and unpacked successfully.")
129
 
@@ -145,7 +145,7 @@ def find_datasets_and_model_ids(root_dir):
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
 
123
  if not os.path.exists(root_dir):
124
  # If root directory doesn't exist, download a zip file and unpack it
125
  print("Root directory doesn't exist. Downloading zip file...")
126
+ url = "https://drive.usercontent.google.com/download?id=1i5UkWikRZGhsbv21ZZSjEZl6-VwNC0lp&export=download&authuser=0&confirm=t&uuid=c33ef625-9ec8-4dbf-bdb0-ad6cabc70a33&at=APZUnTWWJSzU9pV2XV-sMPtbgdgj%3A1711096726305" # Replace with your actual download URL
127
  download_and_extract_zip(url, root_dir)
128
  print("Zip file downloaded and unpacked successfully.")
129
 
 
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 collections.OrderedDict(sorted(datasets.items()))
149
 
150
 
151
  # Main content