Upload 2 files

app.py

@@ -0,0 +1,59 @@
+import gradio as gr
+from PIL import Image
+import random
+import os
+
+# Dummy classifier function (replace with real model logic)
+def classify_parrot(image):
+    # Dummy label list
+    labels = ["Hyacinth Macaw", "Eclectus Parrot", "Cockatiel", "Kakapo", "Budgerigar", "Palm Cockatoo", "Sulphur-crested Cockatoo"]
+    prediction = random.choice(labels)
+    output_img_path = "output_result.png"
+    image.save(output_img_path)
+    return output_img_path, f"Predicted Species: {prediction}"
+
+# Example images (replace with your actual trial images)
+example_images = [
+    "examples/hyacinth.jpg",
+    "examples/eclectus.jpg",
+    "examples/cockatiel.jpg",
+    "examples/kakapo.jpg",
+    "examples/macaw.jpg",
+    "examples/sulphur.jpg",
+    "examples/green_parrot.jpg",
+    "examples/palm_cockatoo.jpg",
+    "examples/yellow_parrot.jpg",
+]
+
+with gr.Blocks(title="Parrot60 Classifier") as demo:
+    gr.Markdown("## 🦜 Parrot60 Classifier")
+    gr.Markdown("This classifier can classify 60 parrot species with 93% accuracy.")
+
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(type="pil", label="Input Image", tool="editor")
+            submit_btn = gr.Button("Submit")
+            clear_btn = gr.Button("Clear")
+        with gr.Column():
+            output_image = gr.Image(label="Output", type="filepath")
+            output_text = gr.Textbox(label="Prediction", lines=1)
+            download_output = gr.File(label="Download Output")
+
+    with gr.Row():
+        gr.Markdown("### 🧪 Examples")
+    with gr.Row():
+        gr.Examples(
+            examples=example_images,
+            inputs=image_input,
+            label="Click to try an example",
+        )
+
+    def run_pipeline(img):
+        result_path, label = classify_parrot(img)
+        return result_path, label, result_path
+
+    submit_btn.click(fn=run_pipeline, inputs=image_input, outputs=[output_image, output_text, download_output])
+    clear_btn.click(fn=lambda: (None, "", None), outputs=[image_input, output_text, download_output])
+
+# Launch app
+demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+rembg==2.0.67
+onnxruntime==1.22.1
+Pillow==11.2.1