Update app.py

app.py

@@ -26,9 +26,9 @@ logger.info(f"OpenCV using {cv2.getNumThreads()} threads out of {cpu_cores} avai
 
 def extract_first_frame(stream_url):
     """
-    Extracts the first available frame from the IP camera stream and returns it as a PIL image.
+    Extracts the current frame from the IP camera stream and returns it as a PIL image.
     """
-    logger.info("Attempting to extract the first frame from the stream...")
+    logger.info("Attempting to extract the current frame from the stream...")
     cap = cv2.VideoCapture(stream_url)
     if not cap.isOpened():
         logger.error("Error: Could not open stream.")
@@ -38,15 +38,15 @@ def extract_first_frame(stream_url):
     cap.release()
 
     if not ret:
-        logger.error("Error: Could not read the first frame.")
-        return None, "Error: Could not read the first frame."
+        logger.error("Error: Could not read the current frame.")
+        return None, "Error: Could not read the current frame."
 
     # Convert the frame to a PIL image
     frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
     pil_image = Image.fromarray(frame_rgb)
 
-    logger.info("First frame extracted successfully.")
-    return pil_image, "First frame extracted successfully."
+    logger.info("Current frame extracted successfully.")
+    return pil_image, "Current frame extracted successfully."
 
 def update_line(image, evt: gr.SelectData):
     """
@@ -296,7 +296,7 @@ with gr.Blocks() as demo:
 
         # Step 2: Select classes to detect
         gr.Markdown("### Step 2: Select Classes to Detect")
-        model = YOLO(model="yolov8n.pt")  # Load the model to get class names
+        model = YOLO(model="yolo12n.pt")  # Load the model to get class names
         class_names = list(model.names.values())  # Get class names
         selected_classes = gr.CheckboxGroup(choices=class_names, label="Select Classes to Detect")