Update app.py

app.py

@@ -9,7 +9,8 @@ model = load_model('traffic_classifier.h5')
 
 def process_image(image):
   img=np.array(image)
-  img=img.convert('RGB')
+  if img.shape[-1] == 4:
+        img = img[:,:,:3]
   img=cv2.resize(img,(30,30))
   img=img/255.0
   img=np.expand_dims(img,axis=0)