Update app.py
Browse files
app.py
CHANGED
@@ -24,11 +24,11 @@ def classify_image(img):
|
|
24 |
return pred_class, dict(zip(categories, map(float,probs)))
|
25 |
|
26 |
# Cell
|
27 |
-
image = gr.
|
28 |
|
29 |
-
set_label = gr.
|
30 |
|
31 |
-
set_prob = gr.
|
32 |
examples = ['test1.jpg', 'test2.jpg', 'test3.jpeg', 'test4.jpeg', 'test5.jpeg', 'test6.jpeg', 'test7.jpeg', 'test8.jpeg', 'test9.jpeg', 'test10.jpeg']
|
33 |
|
34 |
intf = gr.Interface(fn=classify_image,
|
|
|
24 |
return pred_class, dict(zip(categories, map(float,probs)))
|
25 |
|
26 |
# Cell
|
27 |
+
image = gr.Image(shape=(192, 192))
|
28 |
|
29 |
+
set_label = gr.Textbox(label="Predicted Class")
|
30 |
|
31 |
+
set_prob = gr.Label(num_top_classes=2, label="Predicted Probability Per Class")
|
32 |
examples = ['test1.jpg', 'test2.jpg', 'test3.jpeg', 'test4.jpeg', 'test5.jpeg', 'test6.jpeg', 'test7.jpeg', 'test8.jpeg', 'test9.jpeg', 'test10.jpeg']
|
33 |
|
34 |
intf = gr.Interface(fn=classify_image,
|