fix the class_names
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ class_names = []
|
|
| 12 |
|
| 13 |
# Open Cifar class names file and read each line into a list
|
| 14 |
with open('class_names.txt', 'r') as f:
|
| 15 |
-
class_names = [obj.strip() for obj in
|
| 16 |
|
| 17 |
### 2. Model and transforms preparation ###
|
| 18 |
model, model_transforms = create_vitB16_model(num_classes=len(class_names))
|
|
|
|
| 12 |
|
| 13 |
# Open Cifar class names file and read each line into a list
|
| 14 |
with open('class_names.txt', 'r') as f:
|
| 15 |
+
class_names = [obj.strip() for obj in f]
|
| 16 |
|
| 17 |
### 2. Model and transforms preparation ###
|
| 18 |
model, model_transforms = create_vitB16_model(num_classes=len(class_names))
|