Update app.py

app.py

@@ -50,8 +50,9 @@ def generate_journal_with_images(video_path):
         # Make predictions using YOLOv10 on the current frame
         results = model.predict(source=frame_rgb, device=device)
         
-        # Draw bounding boxes on the frame
-        results.render()  # Render the results on the image (this modifies the frame in-place)
+        # 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)
         
         # Save the image with bounding boxes
         frame_filename = os.path.join(output_folder, f"frame_{frame_count}.jpg")
@@ -59,7 +60,7 @@ def generate_journal_with_images(video_path):
         saved_images.append(frame_filename)
         
         # Extract labels (class indices) and map them to class names
-        detected_objects = [model.names[int(box.cls)] for box in results.boxes]
+        detected_objects = [model.names[int(box.cls)] for box in results[0].boxes]  # Access the first result
         
         # Get current timestamp in the video
         timestamp = cap.get(cv2.CAP_PROP_POS_MSEC) / 1000  # Convert ms to seconds
@@ -98,7 +99,7 @@ def display_journal_with_images(video):
     return display_items
 
 with gr.Blocks() as iface:
-    video_input = gr.Video(label="Upload Video")
+    video_input = gr.Video(label="Upload Video", height=300)
     output_gallery = gr.Gallery(label="Generated Daily Journal with Images")
     run_button = gr.Button("Generate Journal")