Update app.py

app.py

@@ -95,9 +95,8 @@ def merge_segments_by_labels(gallery_images, labels_input):
 
 def set_hunyuan_client(request: gr.Request):
     try:
-        x_ip_token = request.headers['x-ip-token']
-        client = Client("tencent/Hunyuan3D-2", headers={"X-IP-Token": x_ip_token})
-        print(x_ip_token, "tencent/Hunyuan3D-2 Ip token")
+        client = Client("tencent/Hunyuan3D-2", headers={"X-IP-Token": request.headers['x-ip-token']})
+        print("tencent/Hunyuan3D-2 Ip token")
         return client
     except:
         print("tencent/Hunyuan3D-2 no token")
@@ -105,9 +104,8 @@ def set_hunyuan_client(request: gr.Request):
 
 def set_vFusion_client(request: gr.Request):
     try:
-        x_ip_token = request.headers['x-ip-token']
-        client = Client("facebook/VFusion3D", headers={"X-IP-Token": x_ip_token})
-        print(x_ip_token, "facebook/VFusion3D Ip token")
+        client = Client("facebook/VFusion3D", headers={"X-IP-Token": request.headers['x-ip-token']})
+        print("facebook/VFusion3D Ip token")
         return client
     except:
         print("facebook/VFusion3D no token")
@@ -129,9 +127,11 @@ def generate_3d_model(client, segment_output, segment_name):
             print(result)
             return result[0]
 
-def generate_3d_model_texture(client, segment_output, segment_name):
+def generate_3d_model_texture(client, segment_output, segment_name, request: gr.Request):
     for i, (image_path, label) in enumerate(segment_output):
         if label == segment_name:
+            client = Client("tencent/Hunyuan3D-2", headers={"X-IP-Token": request.headers['x-ip-token']})
+            print("tencent/Hunyuan3D-2 new client token")
             result = client.predict(
                 caption="",
                 image=handle_file(image_path),
@@ -195,21 +195,8 @@ with gr.Blocks() as demo:
                 hunyuan_button_texture = gr.Button("Hunyuan3D-2 (with texture)")
                 vFusion_button = gr.Button("VFusion3D")
             with gr.Column(scale=5):
-                # trellis_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300)
-                # trellis_output2 = LitModel3D(
-                #     clear_color=[0.1, 0.1, 0.1, 0],  # can adjust background color for better contrast
-                #     label="3D Model Visualization",
-                #     scale=1.0,
-                #     tonemapping="aces",  # can use aces tonemapping for more realistic lighting
-                #     exposure=1.0,        # can adjust exposure to control brightness
-                #     contrast=1.1,        # can slightly increase contrast for better depth
-                #     camera_position=(0, 0, 2),  # will set initial camera position to center the model
-                #     zoom_speed=0.5,      # will adjust zoom speed for better control
-                #     pan_speed=0.5,       # will adjust pan speed for better control
-                #     interactive=True     # this allow users to interact with the model
-                # )
                 trellis_output = gr.Model3D(label="3D Model")
-                # trellis_button.click(generate_3d_model, inputs=[client, segment_output, trellis_input], outputs=trellis_output)
+                
                 hunyuan_button.click(generate_3d_model, inputs=[hunyuan_client, segment_output, trellis_input], outputs=trellis_output)
                 hunyuan_button_texture.click(generate_3d_model_texture, inputs=[hunyuan_client, segment_output, trellis_input], outputs=trellis_output)
                 vFusion_button.click(generate_3d_model2, inputs=[vFusion_client, segment_output, trellis_input], outputs=trellis_output)