Update app.py

app.py

@@ -1,8 +1,14 @@
 import gradio as gr
 import cv2
 import numpy as np
+import base64
 from gradio_client import Client
 
+def np_to_base64(img_np):
+    _, buffer = cv2.imencode('.png', img_np)
+    img_bytes = buffer.tobytes()
+    return "data:image/png;base64," + base64.b64encode(img_bytes).decode()
+
 # -----------------------------
 # Local "try-on" function
 # -----------------------------
@@ -45,14 +51,14 @@ def tryon_to_3d(person_img, garment_img, seed, randomize_seed):
         return None, "Try-on failed"
 
     try:
-        # Convert NumPy arrays to PNG bytes
-        _, tryon_bytes = cv2.imencode('.png', tryon_img)
-        _, garment_bytes = cv2.imencode('.png', garment_img)
+        # Convert images to base64 strings
+        person_base64 = np_to_base64(tryon_img)
+        garment_base64 = np_to_base64(garment_img)
 
-        # Send bytes (Hunyuan3D Gradio API accepts file-like input)
+        # Send base64 images to Hunyuan3D API
         result_3d = hunyuan_client.predict(
-            tryon_bytes.tobytes(),
-            garment_bytes.tobytes(),
+            person_base64,
+            garment_base64,
             fn_index=fn_index_default
         )
         return result_3d, tryon_info