Update app.py

app.py

@@ -1,3 +1,4 @@
+import gradio as gr
 import numpy as np
 from PIL import Image
 import tensorflow as tf
@@ -49,4 +50,20 @@ def get_predictions(image: Image) -> tf.Tensor:
     probs = tf.nn.softmax(prediction['logits'], axis=1)
 
     confidences = {labels[i]: float(probs[0][i]) for i in range(3)}
-    return confidences
+    return confidences
+    
+title = "Simple demo for a Image Classification of the Beans Dataset with HF ViT model"
+
+description = """
+
+"""
+
+demo = gr.Interface(
+    get_predictions,
+    gr.inputs.Image(type="pil"),
+    gr.outputs.Label(num_top_classes=3),
+    allow_flagging="never",
+    title=title,
+)
+
+demo.launch(debug=True)