Update app.py

app.py

@@ -18,7 +18,6 @@ st.write("This app uses Hugging Face Transformers, OpenCV, and Streamlit for det
 # Authorized Car Database
 authorized_cars = {"KA01AB1234", "MH12XY5678", "DL8CAF9090"}  # Dummy data for verification
 
-
 # Detect License Plates
 def detect_license_plate(frame):
     pil_image = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
@@ -30,14 +29,12 @@ def detect_license_plate(frame):
     results = detr_processor.post_process_object_detection(outputs, target_sizes=target_sizes, threshold=0.9)
     return results[0]["boxes"], pil_image
 
-
 # Recognize Text from Plates
 def recognize_text_from_plate(cropped_plate):
     inputs = trocr_processor(images=cropped_plate, return_tensors="pt")
     outputs = trocr_model.generate(**inputs)
     return trocr_processor.batch_decode(outputs, skip_special_tokens=True)[0]
 
-
 # Verify Plate Text
 def verify_plate(plate_text):
     if plate_text in authorized_cars:
@@ -45,15 +42,19 @@ def verify_plate(plate_text):
     else:
         return f"❌ Access Denied: {plate_text}"
 
-
 # Real-Time Video Processing with OpenCV
 def live_feed():
     cap = cv2.VideoCapture(0)  # Open webcam
-    stframe = st.empty()  # Placeholder for video stream
+    if not cap.isOpened():
+        st.error("Unable to access the camera.")
+        return
+
+    stframe = st.image([])  # Placeholder for video feed
 
-    while cap.isOpened():
+    while True:
         ret, frame = cap.read()
         if not ret:
+            st.error("Failed to capture frame from the camera. Exiting...")
             break
 
         # Detect plates
@@ -84,7 +85,6 @@ def live_feed():
     cap.release()
     cv2.destroyAllWindows()
 
-
 # Streamlit UI
 if st.button("Start Camera"):
     live_feed()