Updated pipeline to address the cases where no face is detected

app.py

@@ -76,9 +76,12 @@ if img_file_buffer is not None:
     # To read image file buffer as a PIL Image:
     img = Image.open(img_file_buffer)
     
-    detected_image = Image.fromarray(mtcnn2(img).numpy().transpose(1, 2, 0).astype(np.uint8))
-    st.image(detected_image, caption="Detected Face")
-    
-    embeddings = extract_features(img)
-    bmi = round(lr.predict([embeddings])[0], 2) - 4
-    st.write(f"Your BMI is {bmi}")
+    detected_face = mtcnn2(img)
+    if detected_face is None:
+        st.write("No Face Detected")
+    else:
+        detected_face = Image.fromarray(detected_face.numpy().transpose(1, 2, 0).astype(np.uint8))
+        st.image(detected_face, caption="Detected Face")
+        embeddings = extract_features(img)
+        bmi = round(lr.predict([embeddings])[0], 2) - 4
+        st.write(f"Your BMI is {bmi}")