Spaces:
Runtime error
Runtime error
jervinjosh68
commited on
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
|
|