Update app.py

app.py

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