Update app.py

app.py

@@ -50,13 +50,12 @@ def generate_journal_with_images(video_path):
         # Make predictions using YOLOv10 on the current frame
         results = model.predict(source=frame_rgb, device=device)
         
-        # Loop through the list of results and render bounding boxes
-        for result in results:
-            result.render()  # Render the results on the image (modifies the frame in-place)
+        # Plot bounding boxes and labels on the image
+        annotated_frame = results[0].plot()  # Plot detection results on the frame
         
-        # Save the image with bounding boxes
+        # Save the annotated image
         frame_filename = os.path.join(output_folder, f"frame_{frame_count}.jpg")
-        cv2.imwrite(frame_filename, frame_rgb[:, :, ::-1])  # Convert back to BGR for saving
+        cv2.imwrite(frame_filename, annotated_frame[:, :, ::-1])  # Convert back to BGR for saving
         saved_images.append(frame_filename)
         
         # Extract labels (class indices) and map them to class names