Update app.py

app.py

@@ -50,14 +50,23 @@ def get_predictions(image):
     confidences = {labels[i]: float(probs[0][i]) for i in range(3)}
     return confidences
 
-title = "Simple demo for a Image Classification of the Beans Dataset with HF ViT model"
+with gr.Block() as demo:
+    gr.Markdown("## Simple demo for a Image Classification of the Beans Dataset with HF ViT model")
 
-demo = gr.Interface(
-    get_predictions,
-    gr.inputs.Image(),
-    gr.outputs.Label(num_top_classes=3),
-    allow_flagging="never",
-    title=title,
-)
+    wb_token_if = gr.Textbox("Your Weight & Biases API Key")
 
-demo.launch(debug=True)   
+    with gr.Row():
+        image_if = gr.Image()
+        label_if = gr.Label()
+
+    classify_if = gr.Button()
+
+# demo = gr.Interface(
+#     get_predictions,
+#     gr.inputs.Image(),
+#     gr.outputs.Label(num_top_classes=3),
+#     allow_flagging="never",
+#     title=title,
+# )
+
+demo.launch(debug=True)