Update app.py

app.py

@@ -4,13 +4,25 @@ from transformers import DetrImageProcessor, DetrForObjectDetection, TrOCRProces
 from PIL import Image
 from datetime import datetime
 
+# Ensure all required libraries are installed
+try:
+    import timm  # Required by DETR
+except ImportError:
+    raise ImportError("The 'timm' library is required but not installed. Install it using 'pip install timm'.")
+
 # Load the DETR model for object detection (license plate detection)
-detr_processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
-detr_model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
+try:
+    detr_processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
+    detr_model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
+except Exception as e:
+    raise RuntimeError(f"Error initializing DETR model: {e}")
 
 # Load the TrOCR model for OCR (license plate text recognition)
-trocr_processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-handwritten")
-trocr_model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-base-handwritten")
+try:
+    trocr_processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-handwritten")
+    trocr_model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-base-handwritten")
+except Exception as e:
+    raise RuntimeError(f"Error initializing TrOCR model: {e}")
 
 def detect_license_plate(frame):
     # Convert the frame to a PIL image