fix bugs

app.py

@@ -1,32 +1,31 @@
 from fastai.vision.all import *
 import gradio as gr
-import os
-
-print(gr.__version__)
 
 categories = ("grizzly bear", "black bear", "teddy bear")
-
 learn = load_learner("export.pkl")
 
 def classify_image(img):
     pred, idx, probs = learn.predict(img)
     return dict(zip(categories, map(float, probs)))
 
-image = gr.Image(height=192, width=192)
-label = gr.Label()
-examples = [
-    "examples/bears.jpg",
-    "examples/grizzly.jpg"
-]
+with gr.Blocks() as demo:
+    gr.Markdown("# 🐻 Bear Classifier")
+    gr.Markdown("Classify different types of bears: grizzly, black, or teddy bears.")
+
+    with gr.Row():
+        image_input = gr.Image(height=192, width=192)
+        label_output = gr.Label()
+
+    btn = gr.Button("Classify")
+    btn.click(fn=classify_image, inputs=image_input, outputs=label_output)
 
-demo = gr.Interface(
-    fn=classify_image, 
-    inputs=image, 
-    outputs=label, 
-    examples=examples,
-    title="Bear Classifier",
-    description="Classify different types of bears: grizzly, black, or teddy bears."
-)
+    gr.Examples(
+        examples=[
+            ["examples/bears.jpg"],
+            ["examples/grizzly.jpg"]
+        ],
+        inputs=image_input
+    )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()