update

app.py

@@ -1,3 +1,4 @@
+print('start!', flush=True)
 import gradio as gr
 from models import build_model
 from PIL import Image
@@ -8,6 +9,7 @@ import torch
 from tqdm import trange
 import imageio
 
+print('load!', flush=True)
 checkpoint = 'clevr.pth'
 state = torch.load(checkpoint, map_location='cpu')
 G = build_model(**state['model_kwargs_init']['generator_smooth'])
@@ -19,6 +21,7 @@ G_kwargs= dict(noise_mode='const',
                 fused_modulate=False,
                 impl='cuda',
                 fp16_res=None)
+print('load finish', flush=True)
 
 def trans(x, y, z, length):
     w = h = length