Update app.py

app.py

@@ -57,7 +57,11 @@ def analyze_image(image_path):
 
         # If any objects detected, display labels
         if len(results["labels"]) > 0:
-            detected_objects = [str(label.item()) for label in results["labels"]]
+            # Decode the labels to human-readable objects
+            detected_objects = []
+            for label in results["labels"]:
+                # Map the label to the COCO dataset categories (Detr uses COCO labels by default)
+                detected_objects.append(str(label.item()))  # Get the integer value (COCO class id)
             bot_response = f"Objects detected: {', '.join(detected_objects)}."
         else:
             bot_response = "No objects detected."
@@ -69,6 +73,7 @@ def analyze_image(image_path):
         chat_history.append(("Error during image analysis", error_msg))
         return chat_history
 
+
 # Build the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Ken Chatbot")