File size: 1,036 Bytes
ec15564
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
"""Gradio ZeroGPU entry point for the existing Make-It-Animatable workflow."""
import os
import subprocess
import sys

os.chdir(os.path.dirname(os.path.abspath(__file__)))
subprocess.run(["bash", "install.sh"], check=True)

import spaces
import torch
import torchvision
import torch_cluster
import pytorch3d
import bpy
import gradio

print(
    "STARTUP_VERSIONS "
    f"python={sys.version.split()[0]} torch={torch.__version__} "
    f"torchvision={torchvision.__version__} cuda_wheel={torch.version.cuda} "
    f"torch_cluster={getattr(torch_cluster, '__version__', 'unknown')} "
    f"pytorch3d={pytorch3d.__version__} bpy={bpy.app.version_string} "
    f"gradio={gradio.__version__}", flush=True)
assert torch.__version__.startswith("2.8.0"), "ZeroGPU build did not install Torch 2.8.0"

from app.app import *

init_models()
demo = init_blocks()
demo.launch(server_name="0.0.0.0", server_port=7860,
            allowed_paths=[".", "/data/app/data/Standard Run.fbx", "/data/app/output"],
            show_error=True, ssr_mode=False)