Update app.py

app.py

@@ -48,9 +48,13 @@ import numpy as np
 import torch
 import streamlit as st
 import pygame
+import os
 from ultralytics import YOLO
 from camera_input_live import camera_input_live
 
+# Set environment variable to use dummy audio on Hugging Face Spaces
+os.environ["SDL_AUDIODRIVER"] = "dummy"
+
 # Initialize Pygame mixer for audio alarm
 pygame.mixer.init()
 alarm_sound = "alarm.mp3"  # Ensure you have an alarm.mp3 file in your directory
@@ -104,4 +108,4 @@ if image is not None:
         st.success("✅ No Fire Detected")
         if fire_detected:
             pygame.mixer.music.stop()  # Stop alarm
-        fire_detected = False
+        fire_detected = False