Update processor.py

processor.py

@@ -44,9 +44,9 @@ class_colors = {
     5: (0, 255, 255),  # Yellow (Person)
 }
 
-# Initialize the OCR pipeline
-# Replace 'stepfun-ai/GOT-OCR2_0' with the correct model name if different
-ocr_pipeline = pipeline("image-to-text", model="stepfun-ai/GOT-OCR2_0", trust_remote_code=True, device=0 if torch.cuda.is_available() else -1)
+# Load the OCR model
+processor = AutoProcessor.from_pretrained("stepfun-ai/GOT-OCR2_0", trust_remote_code=True)
+model_ocr = AutoModel.from_pretrained("stepfun-ai/GOT-OCR2_0", trust_remote_code=True).to('cuda')
 
 # Load YOLO model
 # Ensure the path to the model is correct