Update app.py

app.py

@@ -42,13 +42,25 @@ def predict_class(image):
     return predicted_class
 
 # UI Design
+# def classify_image(image):
+#     predicted_class = predict_class(image)
+#     output = f"<h2>Predicted Class: <span style='text-transform:uppercase';>{predicted_class}</span></h2>"
+#     return output
+
 def classify_image(image):
-    predicted_class = predict_class(image)
-    output = f"<h2>Predicted Class: <span style='text-transform:uppercase';>{predicted_class}</span></h2>"
+    results = predict_class(image)
+    output = {}
+    for prediction in results:
+        predicted_label = prediction['label']
+        score = prediction['score']
+        output[predicted_label] = score
     return output
 
+
+
 inputs = gr.inputs.Image(label="Upload an image")
-outputs = gr.outputs.HTML(label="Output")
+# outputs = gr.outputs.HTML()
+outputs = gr.outputs.Label(num_top_classes=10)
 
 title = "<h1 style='text-align: center;'>Image Classifier</h1>"
 description = "Upload an image and get the predicted class."