Sarvamangalak commited on
Commit
f0023ea
·
verified ·
1 Parent(s): 7cbf211

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -9,8 +9,10 @@ import torch
9
  import pathlib
10
  import numpy as np
11
  from urllib.parse import urlparse
12
- from PIL import Image
13
  from transformers import AutoImageProcessor, YolosForObjectDetection, DetrForObjectDetection
 
 
 
14
  from PIL import Image, ImageDraw
15
  from transformers import YolosImageProcessor, YolosForObjectDetection
16
  import easyocr
@@ -68,6 +70,7 @@ model.eval()
68
 
69
  reader = easyocr.Reader(['en'], gpu=False)
70
 
 
71
  # -------------------- Plate Color Classifier --------------------
72
  def classify_plate_color(plate_img):
73
  img = np.array(plate_img)
 
9
  import pathlib
10
  import numpy as np
11
  from urllib.parse import urlparse
 
12
  from transformers import AutoImageProcessor, YolosForObjectDetection, DetrForObjectDetection
13
+ import sqlite3
14
+ import pandas as pd
15
+ import matplotlib.pyplot as plt
16
  from PIL import Image, ImageDraw
17
  from transformers import YolosImageProcessor, YolosForObjectDetection
18
  import easyocr
 
70
 
71
  reader = easyocr.Reader(['en'], gpu=False)
72
 
73
+
74
  # -------------------- Plate Color Classifier --------------------
75
  def classify_plate_color(plate_img):
76
  img = np.array(plate_img)