new models added
Browse files"facebook/convnext-large-224"
            "timm/resnet50.a1_in1k"
            "timm/mobilenetv3_large_100.ra_in1k"
            "trpakov/vit-face-expression"
            "rizvandwiki/gender-classification"
            "#q-future/one-align"
            "LukeJacob2023/nsfw-image-detector"
            "vit-base-patch16-224-in21k"
            "not-lain/deepfake"
            "carbon225/vit-base-patch16-224-hentai"
            "facebook/convnext-base-224-22k-1k"
            "facebook/convnext-large-224"
            "facebook/convnext-tiny-224"
            "nvidia/mit-b0"
            "microsoft/resnet-18"
            "microsoft/swinv2-base-patch4-window16-256"
            "andupets/real-estate-image-classification"
            "timm/tf_efficientnetv2_s.in21k"
            "timm/convnext_tiny.fb_in22k"
            "DunnBC22/vit-base-patch16-224-in21k_Human_Activity_Recognition"
            "FatihC/swin-tiny-patch4-window7-224-finetuned-eurosat-watermark"
            "aalonso-developer/vit-base-patch16-224-in21k-clothing-classifier"
            "RickyIG/emotion_face_image_classification"
            "shadowlilac/aesthetic-shadow"
| @@ -0,0 +1,22 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            {
         | 
| 2 | 
            +
                "workbench.colorCustomizations": {
         | 
| 3 | 
            +
                    "activityBar.activeBackground": "#65c89b",
         | 
| 4 | 
            +
                    "activityBar.background": "#65c89b",
         | 
| 5 | 
            +
                    "activityBar.foreground": "#15202b",
         | 
| 6 | 
            +
                    "activityBar.inactiveForeground": "#15202b99",
         | 
| 7 | 
            +
                    "activityBarBadge.background": "#945bc4",
         | 
| 8 | 
            +
                    "activityBarBadge.foreground": "#e7e7e7",
         | 
| 9 | 
            +
                    "commandCenter.border": "#15202b99",
         | 
| 10 | 
            +
                    "sash.hoverBorder": "#65c89b",
         | 
| 11 | 
            +
                    "statusBar.background": "#42b883",
         | 
| 12 | 
            +
                    "statusBar.foreground": "#15202b",
         | 
| 13 | 
            +
                    "statusBarItem.hoverBackground": "#359268",
         | 
| 14 | 
            +
                    "statusBarItem.remoteBackground": "#42b883",
         | 
| 15 | 
            +
                    "statusBarItem.remoteForeground": "#15202b",
         | 
| 16 | 
            +
                    "titleBar.activeBackground": "#42b883",
         | 
| 17 | 
            +
                    "titleBar.activeForeground": "#15202b",
         | 
| 18 | 
            +
                    "titleBar.inactiveBackground": "#42b88399",
         | 
| 19 | 
            +
                    "titleBar.inactiveForeground": "#15202b99"
         | 
| 20 | 
            +
                },
         | 
| 21 | 
            +
                "peacock.color": "#42b883"
         | 
| 22 | 
            +
            }
         | 
| @@ -20,6 +20,31 @@ MODELS = [ | |
| 20 | 
             
                        "-- Reinstated on testing--",
         | 
| 21 | 
             
                        "microsoft/beit-base-patch16-224-pt22k-ft22k", #Classifição geral
         | 
| 22 | 
             
                        "-- New --"
         | 
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
| 23 |  | 
| 24 | 
             
                    ]
         | 
| 25 |  | 
| @@ -50,13 +75,18 @@ def print_result(result): | |
| 50 |  | 
| 51 | 
             
            def main():
         | 
| 52 | 
             
                st.title("Image Classification")
         | 
| 53 | 
            -
                 | 
|  | |
|  | |
|  | |
| 54 | 
             
                shosen_model = st.selectbox("Select the model to use",  MODELS)
         | 
| 55 |  | 
|  | |
|  | |
|  | |
| 56 | 
             
                if input_image is not None:
         | 
| 57 | 
             
                    image_to_classify = Image.open(input_image)
         | 
| 58 | 
             
                    st.image(image_to_classify, caption="Uploaded Image", use_column_width=True)
         | 
| 59 | 
            -
             | 
| 60 | 
             
                    if st.button("Classify"):
         | 
| 61 | 
             
                        image_to_classify = Image.open(input_image)
         | 
| 62 | 
             
                        classification_obj1 =[]
         | 
|  | |
| 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 |  | 
|  | |
| 75 |  | 
| 76 | 
             
            def main():
         | 
| 77 | 
             
                st.title("Image Classification")
         | 
| 78 | 
            +
                st.write("This is a simple web app to test and compare different image classifier models using Hugging Face's image-classification pipeline.")
         | 
| 79 | 
            +
                st.write("From time to time more models will be added to the list. If you want to add a model, please open an issue on the GitHub repository.")
         | 
| 80 | 
            +
                
         | 
| 81 | 
            +
                st.write("The models available are:")    
         | 
| 82 | 
             
                shosen_model = st.selectbox("Select the model to use",  MODELS)
         | 
| 83 |  | 
| 84 | 
            +
                st.write("Upload an image and click on the 'Classify' button to classify the image.")
         | 
| 85 | 
            +
                input_image = st.file_uploader("Upload Image")
         | 
| 86 | 
            +
                
         | 
| 87 | 
             
                if input_image is not None:
         | 
| 88 | 
             
                    image_to_classify = Image.open(input_image)
         | 
| 89 | 
             
                    st.image(image_to_classify, caption="Uploaded Image", use_column_width=True)
         | 
|  | |
| 90 | 
             
                    if st.button("Classify"):
         | 
| 91 | 
             
                        image_to_classify = Image.open(input_image)
         | 
| 92 | 
             
                        classification_obj1 =[]
         | 
| @@ -0,0 +1,5 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            [{000214A0-0000-0000-C000-000000000046}]
         | 
| 2 | 
            +
            Prop3=19,11
         | 
| 3 | 
            +
            [InternetShortcut]
         | 
| 4 | 
            +
            IDList=
         | 
| 5 | 
            +
            URL=https://huggingface.co/spaces/Nunt/simple_image_classifier
         | 
| @@ -0,0 +1,5 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            [{000214A0-0000-0000-C000-000000000046}]
         | 
| 2 | 
            +
            Prop3=19,11
         | 
| 3 | 
            +
            [InternetShortcut]
         | 
| 4 | 
            +
            IDList=
         | 
| 5 | 
            +
            URL=https://nunt-simple-image-classifier.hf.space/
         |