Update app.py

app.py

@@ -147,9 +147,10 @@ with col4:
 if image_file is not None:
 
     #image=cv2.imread(image_file)
-    image = Image.open(image_file)  
+    image = Image.open(image_file) 
     image=np.asarray(image)
-    image = cv2.cvtColor(image,cv2.COLOR_GRAY2RGB)
+    if len(image_np.shape) == 2:
+    	image = cv2.cvtColor(image,cv2.COLOR_GRAY2RGB)
     st.subheader("Original Image")  
     st.image(image,width=1100)