Update app.py

app.py

@@ -87,9 +87,12 @@ def main():
             # gaussian=gaussian_radius,
             log_visual_freq=1000,
             input='text',
-))
+        ))
     device = th.device()
     generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args, device)
+    latent = torch.randn((1, 512), device=device) 
+    img_out, _, _ = generator([latent])
+    imgs_arr = make_image(img_out)
     iface = gr.Interface(
       fn=predict, 
       inputs='text',