Spaces:
Runtime error
Runtime error
File size: 654 Bytes
11709f2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from urllib.request import urlretrieve
import os
RECOGNITION_MODEL_PATH = 'https://github.com/yMayanand/number_plate_recognition/releases/download/v1.0/ocr_point08.pt'
DETECTION_MODEL_PATH = 'https://github.com/yMayanand/number_plate_recognition/releases/download/v1.0/detection.pt'
os.makedirs('./out', exist_ok=True)
reco_model = './out/ocr_point08.pt'
det_model = './out/detection.pt'
if not os.path.exists(reco_model) or not os.path.exists(det_model):
print('🚀 Downloading Model Weights...')
urlretrieve(RECOGNITION_MODEL_PATH, reco_model)
urlretrieve(DETECTION_MODEL_PATH, det_model)
else:
print('Model Weights already there') |