Update app.py

app.py

@@ -107,7 +107,7 @@ with row3_col5:
 
 # load model
 if 'model' not in st.session_state:
-    session_state.model = YOLO("/home/user/app/models/head_model/best.pt")  # load        
+    st.session_state.model = YOLO("/home/user/app/models/head_model/best.pt")  # load        
 
 #####
 ###
@@ -139,7 +139,7 @@ if st.session_state.image is not None:
     components.html("""<hr style="height:2px;border:none;color:#333;background-color:#333;" /> """)
 
     # predictions
-    results = session_state.model.predict(tiles) # predict
+    results = st.session_state.model.predict(tiles) # predict
     
     
     _, row4_col2, row4_col3, _ = st.columns([1,4,4,1], gap="medium")