Nuno-Tome commited on
Commit
e7f1719
1 Parent(s): ba46c41

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"

.vscode/settings.json ADDED
@@ -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
+ }
app.py CHANGED
@@ -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
- input_image = st.file_uploader("Upload Image")
 
 
 
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 =[]
huggingface.co-spaces-Nunt-simple_image_classifier.url ADDED
@@ -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
nunt-simple-image-classifier.hf.space.url ADDED
@@ -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/