Update UI

app.py

@@ -10,12 +10,13 @@ def predict(img):
     return {labels[i]: float(probs[i]) for i in range(len(labels))}
   
   
-gr.Interface(
-    fn=predict,
-    inputs=gr.Image(height=512, width=512),
-    outputs=gr.Label(num_top_classes=3),
-    examples=[
-        './imgs/dali1.jpeg',
-        './imgs/dali2.jpeg',
-        './imgs/dali3.jpeg',
-    ]).launch()
+with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
+    img = gr.Image()
+    with gr.Row():
+        clear_btn = gr.ClearButton(value="Clear")
+        submit_btn = gr.Button("Predict")
+    output = gr.Label(num_top_classes=3, label="Predictions")
+    submit_btn.click(predict, img, output)
+    clear_btn.add(img)
+
+demo.launch()