Update app.py
Browse files
app.py
CHANGED
|
@@ -9,6 +9,7 @@ model = load_model('traffic_classifier.h5')
|
|
| 9 |
|
| 10 |
def process_image(image):
|
| 11 |
img=np.array(image)
|
|
|
|
| 12 |
img=cv2.resize(img,(30,30))
|
| 13 |
img=img/255.0
|
| 14 |
img=np.expand_dims(img,axis=0)
|
|
|
|
| 9 |
|
| 10 |
def process_image(image):
|
| 11 |
img=np.array(image)
|
| 12 |
+
img=img.convert('RGB')
|
| 13 |
img=cv2.resize(img,(30,30))
|
| 14 |
img=img/255.0
|
| 15 |
img=np.expand_dims(img,axis=0)
|