Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+from io import BytesIO
 from PIL import Image
 from terran.io import open_image
 from terran.vis import display_image, vis_faces
@@ -7,7 +8,10 @@ from terran.face import face_detection
 file= st.file_uploader('Choose an image',type=['png','jpg'])
 if file is not None:
     st.write(file.name)
-#st.image(file)
+    image = Image.open(ByteIO(file))
+    image = np.asarray(image.convert('RGB'))
+    if len(image.shape) == 2:
+        image = np.stack([image] * 3, axis=-1)
     image = open_image(file.name)
     detections = face_detection(image)
     display_image(vis_faces(image, detections))