Nuno-Tome commited on
Commit
047d5b0
·
1 Parent(s): 0f7e74b

no message

Browse files
Files changed (1) hide show
  1. app.py +16 -1
app.py CHANGED
@@ -4,9 +4,24 @@ from PIL import Image
4
  from datasets import load_dataset, Image, list_datasets
5
  from PIL import Image
6
 
 
 
 
 
 
 
7
 
8
- def main():
9
 
 
 
 
 
 
 
 
 
 
 
10
 
11
  if __name__ == "__main__":
12
  main()
 
4
  from datasets import load_dataset, Image, list_datasets
5
  from PIL import Image
6
 
7
+ MODELS = [
8
+ "google/vit-base-patch16-224", #Classifição geral
9
+ "nateraw/vit-age-classifier" #Classifição de idade
10
+ ]
11
+ MAX_N_LABELS = 5
12
+
13
 
 
14
 
15
+ def main():
16
+ st.title("Bulk Image Classification")
17
+ st.markdown("This app uses several 🤗 models to classify images stored in 🤗 datasets.")
18
+ st.write("Soon we will have a dataset template")
19
+ st.write("**Soon we will have dataset selector**")
20
+ st.markdown("The models available are:")
21
+ st.markdown("**PUT IT HERE**")
22
+
23
+
24
+
25
 
26
  if __name__ == "__main__":
27
  main()