Upload gradio_app.py

gradio_app.py

@@ -153,19 +153,16 @@ def build_model_viewer_html(save_folder, height=660, width=790, textured=False):
     else: 
         mesh_file_path = os.path.join(save_folder, f'white_mesh.glb')
 
-
     print(
         f'Find html file {mesh_file_path}, {os.path.exists(mesh_file_path)}')
-
-    
-    # Encode the GLB file as base64 to embed directly in the HTML
-    with open(mesh_file_path, 'rb') as f:
-        encoded = base64.b64encode(f.read()).decode('utf-8')
     
     html = f"""
-    <model-viewer src="data:model/gltf-binary;base64,{encoded}" 
-                  alt="3D Model" camera-controls auto-rotate 
-                  style="height: 690px; width: 500px;">
+    <script type="module" src="https://unpkg.com/@google/model-viewer/dist/model-viewer.min.js"></script>
+    <model-viewer src="{mesh_file_path}" 
+              alt="3D Model" 
+              camera-controls 
+              auto-rotate 
+              style="height: 690px; width: 500px;">
     </model-viewer>
     """
 
@@ -424,7 +421,12 @@ def build_app():
 
     """
 
-    with gr.Blocks(theme=gr.themes.Base(), title='Hunyuan-3D-2.0', analytics_enabled=False, css=custom_css) as demo:
+    custom_js = """
+    <script type="module" src="https://unpkg.com/@google/model-viewer/dist/model-viewer.min.js"></script>
+    """
+
+
+    with gr.Blocks(theme=gr.themes.Base(), title='Hunyuan-3D-2.0', analytics_enabled=False, css=custom_css, js=custom_js) as demo:
         gr.HTML(title_html)
 
         with gr.Row():