bug fix
Browse files
app.py
CHANGED
|
@@ -19,32 +19,32 @@ MODELS = [
|
|
| 19 |
"microsoft/swin-tiny-patch4-window7-224",#Classifição geral
|
| 20 |
"-- Reinstated on testing--",
|
| 21 |
"microsoft/beit-base-patch16-224-pt22k-ft22k", #Classifição geral
|
| 22 |
-
"-- New --"
|
| 23 |
-
"-- Still in the testing process --"
|
| 24 |
-
"facebook/convnext-large-224"
|
| 25 |
-
"timm/resnet50.a1_in1k"
|
| 26 |
-
"timm/mobilenetv3_large_100.ra_in1k"
|
| 27 |
-
"trpakov/vit-face-expression"
|
| 28 |
-
"rizvandwiki/gender-classification"
|
| 29 |
-
"#q-future/one-align"
|
| 30 |
-
"LukeJacob2023/nsfw-image-detector"
|
| 31 |
-
"vit-base-patch16-224-in21k"
|
| 32 |
-
"not-lain/deepfake"
|
| 33 |
-
"carbon225/vit-base-patch16-224-hentai"
|
| 34 |
-
"facebook/convnext-base-224-22k-1k"
|
| 35 |
-
"facebook/convnext-large-224"
|
| 36 |
-
"facebook/convnext-tiny-224"
|
| 37 |
-
"nvidia/mit-b0"
|
| 38 |
-
"microsoft/resnet-18"
|
| 39 |
-
"microsoft/swinv2-base-patch4-window16-256"
|
| 40 |
-
"andupets/real-estate-image-classification"
|
| 41 |
-
"timm/tf_efficientnetv2_s.in21k"
|
| 42 |
-
"timm/convnext_tiny.fb_in22k"
|
| 43 |
-
"DunnBC22/vit-base-patch16-224-in21k_Human_Activity_Recognition"
|
| 44 |
-
"FatihC/swin-tiny-patch4-window7-224-finetuned-eurosat-watermark"
|
| 45 |
-
"aalonso-developer/vit-base-patch16-224-in21k-clothing-classifier"
|
| 46 |
-
"RickyIG/emotion_face_image_classification"
|
| 47 |
-
"shadowlilac/aesthetic-shadow"
|
| 48 |
]
|
| 49 |
|
| 50 |
def classify(image, model):
|
|
|
|
| 19 |
"microsoft/swin-tiny-patch4-window7-224",#Classifição geral
|
| 20 |
"-- Reinstated on testing--",
|
| 21 |
"microsoft/beit-base-patch16-224-pt22k-ft22k", #Classifição geral
|
| 22 |
+
"-- New --",
|
| 23 |
+
"-- Still in the testing process --",
|
| 24 |
+
"facebook/convnext-large-224", #Classifição geral
|
| 25 |
+
"timm/resnet50.a1_in1k", #Classifição geral
|
| 26 |
+
"timm/mobilenetv3_large_100.ra_in1k", #Classifição geral
|
| 27 |
+
"trpakov/vit-face-expression", #Classifição de expressão facial
|
| 28 |
+
"rizvandwiki/gender-classification", #Classifição de gênero
|
| 29 |
+
"#q-future/one-align", #Classifição geral
|
| 30 |
+
"LukeJacob2023/nsfw-image-detector", #Classifição NSFW
|
| 31 |
+
"vit-base-patch16-224-in21k", #Classifição geral
|
| 32 |
+
"not-lain/deepfake", #Classifição deepfake
|
| 33 |
+
"carbon225/vit-base-patch16-224-hentai", #Classifição hentai
|
| 34 |
+
"facebook/convnext-base-224-22k-1k", #Classifição geral
|
| 35 |
+
"facebook/convnext-large-224", #Classifição geral
|
| 36 |
+
"facebook/convnext-tiny-224",#Classifição geral
|
| 37 |
+
"nvidia/mit-b0", #Classifição geral
|
| 38 |
+
"microsoft/resnet-18", #Classifição geral
|
| 39 |
+
"microsoft/swinv2-base-patch4-window16-256", #Classifição geral
|
| 40 |
+
"andupets/real-estate-image-classification", #Classifição de imóveis
|
| 41 |
+
"timm/tf_efficientnetv2_s.in21k", #Classifição geral
|
| 42 |
+
"timm/convnext_tiny.fb_in22k",
|
| 43 |
+
"DunnBC22/vit-base-patch16-224-in21k_Human_Activity_Recognition", #Classifição de atividade humana
|
| 44 |
+
"FatihC/swin-tiny-patch4-window7-224-finetuned-eurosat-watermark", #Classifição geral
|
| 45 |
+
"aalonso-developer/vit-base-patch16-224-in21k-clothing-classifier", #Classifição de roupas
|
| 46 |
+
"RickyIG/emotion_face_image_classification", #Classifição de emoções
|
| 47 |
+
"shadowlilac/aesthetic-shadow" #Classifição de estética
|
| 48 |
]
|
| 49 |
|
| 50 |
def classify(image, model):
|