Update app.py

app.py

@@ -27,7 +27,7 @@ if st.button("Extract Head"):
             result = extract_head(uploaded_file)
             
             # Extract image URL from JSON response
-            head_url = result.get("message", {}).get("head_image_url")
+            head_url = result.get("imageURL")
             if head_url:
                 st.image(head_url, caption="Extracted Head", use_container_width=True)
                 
@@ -42,4 +42,4 @@ if st.button("Extract Head"):
                 st.error("Failed to extract head. No image URL found in response.")
                 st.write("Response Data:", result)  # Debugging: Show response data
     else:
-        st.warning("Please upload an image.")
+        st.warning("Please upload an image.")