Update app.py

app.py

@@ -2,7 +2,8 @@ import streamlit as st
 # from transformers import pipeline
 from deepface import DeepFace
 import numpy as np
-from PIL import Image
+# import custom helper functions
+from backend import check_image_rotation
 
 # pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
 
@@ -16,8 +17,8 @@ if file_name is not None:
     # make two columns
     col1, col2 = st.columns(2)
 
-    # capture image
-    image = Image.open(file_name)
+    # capture image with intended rotation
+    image = check_image_rotation(file_name)
 
     # display image in left column
     col1.image(image, use_column_width=True)