Update app.py

app.py

@@ -44,7 +44,7 @@ uploaded_file = st.file_uploader("Upload your selfie", type=["jpg", "png", "jpeg
 
 if uploaded_file is not None:
     # Show uploaded image
-    st.image(uploaded_file, caption="Uploaded Image", use_column_width=True)
+    st.image(uploaded_file, caption="Uploaded Image", use_container_width=True)
     st.info("Detecting emotion... please wait...")
 
     # Save file locally
@@ -58,15 +58,7 @@ if uploaded_file is not None:
         api_name="/preprocess_image_and_predict"
     )
 
-    face_image, heatmap_image, confidences = result
-
-    # Show face & heatmap
-    st.image(face_image, caption="Detected Face", use_column_width=True)
-    st.image(heatmap_image, caption="Heatmap", use_column_width=True)
-
-    # Show probabilities
-    st.subheader("Emotion Probabilities")
-    st.json(confidences)
+    _, _, confidences = result
 
     # Use top label directly
     dominant = confidences["label"]