jervinjosh68 commited on
Commit
9729ed8
1 Parent(s): 0e76714

minor changes to app.py

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. gradio_queue.db +0 -0
app.py CHANGED
@@ -17,7 +17,7 @@ def predict(im):
17
  proba = output.logits.softmax(1)
18
 
19
  # Predicted Classes
20
- #preds = proba.argmax(1)
21
  values, indices = torch.topk(proba, k=5)
22
 
23
 
@@ -38,7 +38,7 @@ examples = ["stock_baby.webp","stock_teen.webp","stock_guy.jpg","stock_old_woman
38
  gr.Interface(
39
  predict,
40
  inputs,
41
- outputs = 'Estimated Age Bracket',
42
  title=title,
43
  description=description,
44
  article=article,
 
17
  proba = output.logits.softmax(1)
18
 
19
  # Predicted Classes
20
+ preds = proba.argmax(1)
21
  values, indices = torch.topk(proba, k=5)
22
 
23
 
 
38
  gr.Interface(
39
  predict,
40
  inputs,
41
+ outputs = 'label',
42
  title=title,
43
  description=description,
44
  article=article,
gradio_queue.db CHANGED
Binary files a/gradio_queue.db and b/gradio_queue.db differ