Spaces:
Running
Running
Commit
•
9729ed8
1
Parent(s):
0e76714
minor changes to app.py
Browse files- app.py +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 |
-
|
| 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 = '
|
| 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
|