change the way labels look (make them spaced out and capitalized)
Browse files
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)
|