Update app.py

app.py

@@ -25,9 +25,10 @@ st.title("GrayScale to Colorized Image Pix2Pix")
 day_inp = st.file_uploader("Grayscale image input")
 
 if day_inp is not None:
-  img = cv2.imread(day_inp.name)
+  file_bytes = day_inp.read()
+  img = cv2.imdecode(np.frombuffer(file_bytes, np.uint8), cv2.IMREAD_GRAYSCALE)
   img = cv2.resize(img, (256, 256))
-  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+  img = cv2.cvtColor(img, cv2.COLOR_GRAY2RGB)
   img = np.array(img)
   pred = model_out('colorizer.h5', img)
   st.image(img, caption="Uploaded Image")