Update app.py

app.py

@@ -22,7 +22,8 @@ def visualize_texture():
     # Apply the texture to the original mesh
     textured_mesh = trimesh.Trimesh(vertices=mesh.vertices, faces=mesh.faces, visual=color_visuals, validate=True, process= False)
     
-    return textured_mesh.export('trainm.glb')
+    modified_model = textured_mesh.export('trainm.glb')
+    return modified_model
     
 
 with gr.Blocks() as app:
@@ -30,6 +31,6 @@ with gr.Blocks() as app:
     original_model = gr.Model3D('train.glb', label="Original Model")
     modified_model = gr.Model3D(label="Textured Model")
     button = gr.Button("Visualize Texture")
-    button.click(visualize_texture, outputs=[original_model, modified_model])
+    button.click(visualize_texture, outputs=[modified_model])
 
 app.launch()