xtlyxt commited on
Commit
2c4ccb2
·
verified ·
1 Parent(s): a8d155f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ if st.button("Predict Emotions") and selected_images:
51
  st.session_state['results'] = [pipe(image) for image in selected_images]
52
 
53
  # Generate DataFrame from results
54
- if st.button("Generate DataFrame"):
55
  # Access the results, image names, and sizes from the session state
56
  results = st.session_state['results']
57
  image_names = st.session_state['image_names']
 
51
  st.session_state['results'] = [pipe(image) for image in selected_images]
52
 
53
  # Generate DataFrame from results
54
+ if st.button("Generate HeatMap & DataFrame"):
55
  # Access the results, image names, and sizes from the session state
56
  results = st.session_state['results']
57
  image_names = st.session_state['image_names']