code cleanup

app.py

@@ -15,11 +15,6 @@ from facenet_pytorch import MTCNN
 model_file_path = 'models/bmi.h5'
 model = tf.keras.models.load_model(model_file_path)
 
-# mtcnn2 = MTCNN(
-#         image_size=160, margin=40, min_face_size=20,
-#         thresholds=[0.6, 0.7, 0.7], factor=0.709, post_process=False
-#     )
-
 # Preprocess the images for VGG16
 def preprocess_image(img_path):
     img = image.load_img(img_path, target_size = (224, 224))
@@ -37,15 +32,10 @@ def main():
     img_file_buffer = st.camera_input("Take a picture")
 
     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 = preprocess_image(img_file_buffer)
         bmi = model.predict(embeddings)
-        st.write(f"Your BMI is {bmi}")
+        st.write(f"Your BMI is {bmi[0][0]}")
 
 if __name__ == '__main__':
     main()