no message
Browse files
app.py
CHANGED
@@ -6,12 +6,13 @@ from PIL import Image
|
|
6 |
|
7 |
MODELS = [
|
8 |
"google/vit-base-patch16-224", #Classifição geral
|
9 |
-
"nateraw/vit-age-classifier"
|
10 |
-
|
11 |
]
|
12 |
DATASETS = [
|
13 |
"Nunt/testedata",
|
14 |
-
"Nunt/backup_leonardo"
|
|
|
15 |
]
|
16 |
MAX_N_LABELS = 5
|
17 |
SPLIT_TO_CLASSIFY = 'pasta'
|
|
|
6 |
|
7 |
MODELS = [
|
8 |
"google/vit-base-patch16-224", #Classifição geral
|
9 |
+
"nateraw/vit-age-classifier" #Classifição de idade
|
10 |
+
|
11 |
]
|
12 |
DATASETS = [
|
13 |
"Nunt/testedata",
|
14 |
+
"Nunt/backup_leonardo",
|
15 |
+
"Nunt/backup_leonardo_2024-02-06"
|
16 |
]
|
17 |
MAX_N_LABELS = 5
|
18 |
SPLIT_TO_CLASSIFY = 'pasta'
|