ahasera commited on
Commit
ad928cc
·
verified ·
1 Parent(s): 7e180bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ def load_models():
22
  global model_vehicle, model_plate, reader_easyocr, pipeline_kerasocr, processor_trocr, model_trocr
23
  model_vehicle = YOLO('models/yolov8n.pt')
24
  model_plate = YOLO('models/best.pt')
25
- reader_easyocr = easyocr.Reader(['en'], gpu=True)
26
  pipeline_kerasocr = keras_ocr.pipeline.Pipeline()
27
  processor_trocr = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
28
  model_trocr = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten')
 
22
  global model_vehicle, model_plate, reader_easyocr, pipeline_kerasocr, processor_trocr, model_trocr
23
  model_vehicle = YOLO('models/yolov8n.pt')
24
  model_plate = YOLO('models/best.pt')
25
+ reader_easyocr = easyocr.Reader(['en'], gpu=False)
26
  pipeline_kerasocr = keras_ocr.pipeline.Pipeline()
27
  processor_trocr = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
28
  model_trocr = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten')