Update app.py

app.py

@@ -95,7 +95,10 @@ def process_image(img, generate_image=True):
     predicted_class, _, probs = learn.predict(img)
     print(f"Prediction complete: {predicted_class}")
 
-    classification_results = dict(zip(learn.dls.vocab, map(float, probs)))
+    # Format classification results for Gradio Label
+    classification_results = {
+        predicted_class: float(probs[learn.dls.vocab.o2i[predicted_class]])
+    }
 
     # Get Wikipedia link
     wiki_url = search_terms_wikipedia.get(predicted_class, "No Wikipedia entry found.")
@@ -163,8 +166,8 @@ with gr.Blocks() as demo:
         examples=example_images,
         inputs=input_image,
         examples_per_page=6,
-        fn=lambda img: process_image(img, generate_image=False),
-        outputs=[label_output, generated_image, wiki_output, endangerment_output]
+        fn=None,
+        outputs=None
     )
 
     input_image.change(