Update app.py

app.py

@@ -8,10 +8,9 @@ pipe = pipeline("image-classification", "dima806/medicinal_plants_image_detectio
 def image_classifier(image):
     # Perform image classification
     outputs = pipe(image)
-    results = {}
-    for result in outputs:
-        results[result['label']] = result['score']
-    return results
+    # Get the label of the first result
+    output_text = outputs[0]['label']
+    return output_text
 
 # Define app title and description with HTML formatting
 title = "<h1 style='text-align: center; color: #4CAF50;'>Image Classification</h1>"
@@ -37,9 +36,7 @@ custom_css = """
             """
 
 # Launch the Gradio interface with custom HTML and CSS
-demo = gr.Interface(fn=image_classifier, inputs=gr.Image(type="pil"), outputs="text", title=title, description=description, 
+demo = gr.Interface(fn=image_classifier, inputs=gr.Image(type="pil"), outputs="textbox", title=title, description=description, 
                      theme="gstaff/sketch", css=custom_css,
                    )
 demo.launch()
-
-