Update app.py
Browse files
app.py
CHANGED
@@ -27,8 +27,8 @@ def predict(image):
|
|
27 |
return {f"Class {i}": label for i, label in enumerate(top_5)}
|
28 |
|
29 |
# Define Gradio interface
|
30 |
-
image = gr.
|
31 |
-
label = gr.
|
32 |
|
33 |
interface = gr.Interface(
|
34 |
fn=predict,
|
|
|
27 |
return {f"Class {i}": label for i, label in enumerate(top_5)}
|
28 |
|
29 |
# Define Gradio interface
|
30 |
+
image = gr.components.Image(type="pil")
|
31 |
+
label = gr.components.Label(num_top_classes=5)
|
32 |
|
33 |
interface = gr.Interface(
|
34 |
fn=predict,
|