Update app.py

app.py

@@ -11,7 +11,7 @@ import imageio_ffmpeg as ffmpeg
 
 # Page config first
 st.set_page_config(
-    page_title="WildfireWatch: AI Detection",
+    page_title="Fire Watch: AI model detection of fire and smoke",
     page_icon="🔥",
     layout="wide",
     initial_sidebar_state="expanded"
@@ -29,7 +29,7 @@ for key in ["processed_frames", "slider_value", "processed_video", "start_time"]
 with st.sidebar:
     st.header("Upload & Settings")
     source_file = st.file_uploader("Upload image/video", type=["jpg", "jpeg", "png", "bmp", "webp", "mp4"])
-    confidence = float(st.slider("Confidence Threshold", 25, 100, 40)) / 100
+    confidence = float(st.slider("Confidence Threshold", 10, 100, 20)) / 100
     fps_options = {
         "Original FPS": None,
         "3 FPS": 3,
@@ -40,13 +40,13 @@ with st.sidebar:
         "1 frame/30s": 0.0333
     }
     video_option = st.selectbox("Output Frame Rate", list(fps_options.keys()))
-    process_button = st.button("Detect Wildfire")
+    process_button = st.button("Detect fire and smoke")
     progress_bar = st.progress(0)
     progress_text = st.empty()
     download_slot = st.empty()
 
 # Main page
-st.title("WildfireWatch: AI-Powered Detection")
+st.title("Fire Watch: AI-Powered Fire Detection")
 col1, col2 = st.columns(2)
 with col1:
     st.image("https://huggingface.co/spaces/tstone87/ccr-colorado/resolve/main/Fire_1.jpeg", use_column_width=True)
@@ -115,6 +115,9 @@ if process_button and source_file and model:
             if frame_count % sample_interval == 0:
                 res = model.predict(frame, conf=confidence)
                 processed_frame = res[0].plot()[:, :, ::-1]
+                # Ensure frame is C-contiguous
+                if not processed_frame.flags['C_CONTIGUOUS']:
+                    processed_frame = np.ascontiguousarray(processed_frame)
                 st.session_state.processed_frames.append(processed_frame)
                 
                 processed_count += 1
@@ -140,14 +143,13 @@ if process_button and source_file and model:
         
         if st.session_state.processed_frames:
             out_path = tempfile.NamedTemporaryFile(suffix='.mp4', delete=False).name
-            # Use imageio-ffmpeg instead of cv2 for more reliable MP4 output
             writer = ffmpeg.write_frames(
                 out_path,
                 (width, height),
                 fps=output_fps or orig_fps,
-                codec='libx264',  # Widely supported codec
-                pix_fmt_in='bgr24',  # Match OpenCV frame format
-                pix_fmt_out='yuv420p'  # Standard for MP4
+                codec='libx264',
+                pix_fmt_in='bgr24',
+                pix_fmt_out='yuv420p'
             )
             writer.send(None)  # Initialize writer