Update app.py

app.py

@@ -28,4 +28,17 @@ def predict_animal(image):
     return {class_names[i]: float(probs[i]) for i in range(len(class_names))}
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🐾 Animal Type Class
+    gr.Markdown("# 🐾 Animal Type Classifier")
+    gr.Markdown("Upload an image of an animal below and get predictions for butterflies, chickens, elephants, horses, spiders, or squirrels.")
+
+    with gr.Row():
+        img_input = gr.Image(type="pil", label="Upload Animal Image")
+        label_output = gr.Label(num_top_classes=6, label="Prediction Scores")
+
+    predict_button = gr.Button("Classify Animal")
+    predict_button.click(fn=predict_animal, inputs=img_input, outputs=label_output)
+
+    gr.Markdown("Developed with Ultralytics YOLO and Gradio framework.")
+
+if __name__ == "__main__":
+    demo.launch()