Update hg_app.py

hg_app.py

@@ -115,7 +115,11 @@ def build_model_viewer_html(save_folder, height=660, width=790, textured=False):
         {iframe_tag}
         </div>
     """
-
+def get_texgen_worker():
+    if not hasattr(get_texgen_worker, 'worker'):
+        from hy3dgen.texgen import Hunyuan3DPaintPipeline
+        get_texgen_worker.worker = Hunyuan3DPaintPipeline.from_pretrained('tencent/Hunyuan3D-2')
+    return get_texgen_worker.worker
 
 @spaces.GPU(duration=100)
 def _gen_shape(
@@ -181,6 +185,7 @@ def _gen_shape(
     torch.cuda.empty_cache()
     return mesh, save_folder, image
 
+    
 @spaces.GPU(duration=150)
 def generation_all(
     caption: str,
@@ -205,6 +210,8 @@ def generation_all(
     path = export_mesh(mesh, save_folder, textured=False)
     model_viewer_html = build_model_viewer_html(save_folder, height=596, width=700)
 
+    # Initialize texgen_worker within the function
+    texgen_worker = get_texgen_worker()
     textured_mesh = texgen_worker(mesh, image)
     path_textured = export_mesh(textured_mesh, save_folder, textured=True)
     model_viewer_html_textured = build_model_viewer_html(save_folder, height=596, width=700, textured=True)