Update app.py
Browse files
app.py
CHANGED
@@ -18,8 +18,8 @@ def predict(image):
|
|
18 |
return {f"Class {i}": prob.item() for i, prob in enumerate(probs[0])}
|
19 |
|
20 |
# Define Gradio interface
|
21 |
-
image = gr.
|
22 |
-
label = gr.
|
23 |
|
24 |
interface = gr.Interface(
|
25 |
fn=predict,
|
|
|
18 |
return {f"Class {i}": prob.item() for i, prob in enumerate(probs[0])}
|
19 |
|
20 |
# Define Gradio interface
|
21 |
+
image = gr.components.Image(type="pil")
|
22 |
+
label = gr.components.Label(num_top_classes=5)
|
23 |
|
24 |
interface = gr.Interface(
|
25 |
fn=predict,
|