update

.ipynb_checkpoints/app-checkpoint.py

@@ -9,7 +9,7 @@ def get_methods_and_arch(dataset):
     archs   = []
     for column in columns:
         methods.append(column.split('_')[0])
-        archs.append(column.split('_')[1])
+        archs.append('_'.join(column.split('_')[1:-2]))
     return list(set(methods)),list(set(archs))
 
 def get_columns(arch,method):
@@ -77,3 +77,5 @@ if __name__ == '__main__':
         demo.launch()
 
 
+
+

app.py

@@ -9,7 +9,7 @@ def get_methods_and_arch(dataset):
     archs   = []
     for column in columns:
         methods.append(column.split('_')[0])
-        archs.append(column.split('_')[1])
+        archs.append('_'.join(column.split('_')[1:-2]))
     return list(set(methods)),list(set(archs))
 
 def get_columns(arch,method):
@@ -77,3 +77,5 @@ if __name__ == '__main__':
         demo.launch()
 
 
+
+