jervinjosh68 commited on
Commit
0e76714
1 Parent(s): 725b25f

minor changes to app.py and requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +0 -1
app.py CHANGED
@@ -32,13 +32,13 @@ inputs = [
32
 
33
  title = "ViT-Age-Classification"
34
  description = "ViT-Age-Classification is used to categorize an individual's age using images"
35
- article = " <a href='https://huggingface.co/nateraw/vit-age-classifier'>Model Repo on Hugging Face Model Hub</a>"
36
  examples = ["stock_baby.webp","stock_teen.webp","stock_guy.jpg","stock_old_woman.jpg"]
37
 
38
  gr.Interface(
39
  predict,
40
  inputs,
41
- outputs = 'label',
42
  title=title,
43
  description=description,
44
  article=article,
 
32
 
33
  title = "ViT-Age-Classification"
34
  description = "ViT-Age-Classification is used to categorize an individual's age using images"
35
+ article = " <a href='https://huggingface.co/nateraw/vit-age-classifier'>ViT Age Classification Model Repo on Hugging Face Model Hub</a>"
36
  examples = ["stock_baby.webp","stock_teen.webp","stock_guy.jpg","stock_old_woman.jpg"]
37
 
38
  gr.Interface(
39
  predict,
40
  inputs,
41
+ outputs = 'Estimated Age Bracket',
42
  title=title,
43
  description=description,
44
  article=article,
requirements.txt CHANGED
@@ -1,5 +1,4 @@
1
  transformers
2
  gradio
3
- requests
4
  Pillow
5
  torch
 
1
  transformers
2
  gradio
 
3
  Pillow
4
  torch