Update app.py

app.py

@@ -183,6 +183,14 @@ def predict(image):
         results = model.predict(image)  # Perform object detection
         print("YOLO prediction completed.")
 
+        # Print detected class names
+        detected_classes = results.names  # Get class names for each detected object
+        print("Detected classes:", detected_classes)
+
+        # Print the names of the detected classes
+        for i, name in enumerate(detected_classes):
+            print(f"Detected object {i+1}: {name}")
+
         # Render bounding boxes on the image
         results.render()
         print("Bounding boxes rendered.")
@@ -209,3 +217,4 @@ iface = gr.Interface(
 # Launch the Gradio app
 iface.launch()
 
+