vapit commited on
Commit
1f6f643
·
1 Parent(s): 759c19b

change the way labels look (make them spaced out and capitalized)

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -14,6 +14,8 @@ labels = labels_csv['breed']
14
  labels = np.array(labels)
15
  unique_labels = np.unique(labels)
16
 
 
 
17
  ### 2. Model and transforms preparation ###
18
  model, model_transforms = create_model(num_classes=len(unique_labels))
19
  model = torch.compile(model)
 
14
  labels = np.array(labels)
15
  unique_labels = np.unique(labels)
16
 
17
+ unique_labels = [' '.join([word.capitalize() for word in label.split('_')]) for label in unique_labels]
18
+
19
  ### 2. Model and transforms preparation ###
20
  model, model_transforms = create_model(num_classes=len(unique_labels))
21
  model = torch.compile(model)