Update app.py

app.py

@@ -8,20 +8,19 @@ model = YOLO("best.pt")  # Ensure the path to your model is correct
 
 # Set the title of the app
 st.title("Live Fire Detection App")
-
-webcamera = cv2.VideoCapture(0)
-# webcamera.set(cv2.CAP_PROP_FRAME_WIDTH, 1920)
-# webcamera.set(cv2.CAP_PROP_FRAME_HEIGHT, 1080)
+cap = cv2.VideoCapture(0)
+cap.set(3, 640)
+cap.set(4, 480)
 
 while True:
-    success, frame = webcamera.read()
-    
-    results = model.track(frame, classes=0, conf=0.8, imgsz=480)
-    cv2.putText(frame, f"Total: {len(results[0].boxes)}", (50, 50), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2, cv2.LINE_AA)
-    cv2.imshow("Live Camera", results[0].plot())
+    ret, img= cap.read()
+    cv2.imshow('Webcam', img)
 
     if cv2.waitKey(1) == ord('q'):
         break
 
-webcamera.release()
-cv2.destroyAllWindows()
+cap.release()
+cv2.destroyAllWindows(
+
+
+