Update app.py

app.py

@@ -142,8 +142,7 @@ if image_file is not None:
 
     image_original = Image.open(image_file) 
     image=np.asarray(image_original)
-    if len(image.shape) == 2:
-    	image = cv2.cvtColor(image,cv2.COLOR_GRAY2RGB)
+  	image = convert_rgb(image)
     st.subheader("Original Image")  
     st.image(image,width=1100)
 
@@ -182,7 +181,7 @@ if image_file is not None:
         st.image(img,width=1100)
     
     image=np.asarray(image_original)
-    image = cv2.cvtColor(image,cv2.COLOR_GRAY2RGB)
+    image = convert_rgb(image)
     if image.shape[1] < 3000:
         image = cv2.resize(image,(3100,1150),interpolation=cv2.INTER_LANCZOS4)
     predicted=cv2.imread("predict.png")