Update app.py

app.py

@@ -242,15 +242,6 @@ MAX_SEED = 1e7
 title = "## AI 3D Model Generator"
 description = "Our Image-to-3D Generator transforms your 2D photos into stunning, AI generated 3D models—ready for games, AR/VR, or 3D printing. Our AI 3D Modeling is based on Hunyuan 2.0. Check more in [imgto3d.ai](https://www.imgto3d.ai)."
 
-# 顶层哨兵函数，HF 扫描用
-@spaces.GPU(duration=1)
-def _gpu_check():
-    import torch
-    if torch.cuda.is_available():
-        return f"GPU ready: {torch.cuda.get_device_name(0)}"
-    else:
-        return "No GPU available"
-        
 with gr.Blocks().queue() as demo:
     gr.Markdown(title)
     gr.Markdown(description)
@@ -297,18 +288,19 @@ with gr.Blocks().queue() as demo:
         outputs=[html_export_mesh,file_export, glbPath_output, objPath_output]
     )    
 
- # https://discuss.huggingface.co/t/how-to-serve-an-html-file/33921/2
-# create a FastAPI app
-app = FastAPI()
-# create a static directory to store the static files
-static_dir = Path(SAVE_DIR).absolute()
-static_dir.mkdir(parents=True, exist_ok=True)
-app.mount("/static", StaticFiles(directory=static_dir, html=True), name="static")
-shutil.copytree('./assets/env_maps', os.path.join(static_dir, 'env_maps'), dirs_exist_ok=True)
-
-if args.low_vram_mode:
-    torch.cuda.empty_cache()
-
-app = gr.mount_gradio_app(app, demo, path="/")
-# demo.launch()
-uvicorn.run(app, host=args.host, port=args.port)
+if __name__ == "__main__":
+    # https://discuss.huggingface.co/t/how-to-serve-an-html-file/33921/2
+    # create a FastAPI app
+    app = FastAPI()
+    # create a static directory to store the static files
+    static_dir = Path(SAVE_DIR).absolute()
+    static_dir.mkdir(parents=True, exist_ok=True)
+    app.mount("/static", StaticFiles(directory=static_dir, html=True), name="static")
+    shutil.copytree('./assets/env_maps', os.path.join(static_dir, 'env_maps'), dirs_exist_ok=True)
+
+    if args.low_vram_mode:
+        torch.cuda.empty_cache()
+
+    app = gr.mount_gradio_app(app, demo, path="/")
+    # demo.launch()
+    uvicorn.run(app, host=args.host, port=args.port)