HongFangzhou commited on
Commit
0a91ae4
β€’
1 Parent(s): f5d98e9

add source codes

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,4 +1,3 @@
1
- import pytorch_lightning as pl
2
  import os
3
  import sys
4
  import cv2
@@ -14,6 +13,7 @@ import numpy as np
14
  import gradio as gr
15
  from tqdm import tqdm
16
  import imageio.v2 as imageio
 
17
  from omegaconf import OmegaConf
18
  from safetensors.torch import load_file
19
  from huggingface_hub import hf_hub_download
@@ -353,4 +353,4 @@ with block:
353
  download = gr.File(label="Download mesh", file_count="single", height=100)
354
  gr.on([btn_stage2.click], infer_stage2, inputs=[text, dropdown, seed, global_info], outputs=[gallery, download])
355
 
356
- block.launch(share=True)
 
 
1
  import os
2
  import sys
3
  import cv2
 
13
  import gradio as gr
14
  from tqdm import tqdm
15
  import imageio.v2 as imageio
16
+ import pytorch_lightning as pl
17
  from omegaconf import OmegaConf
18
  from safetensors.torch import load_file
19
  from huggingface_hub import hf_hub_download
 
353
  download = gr.File(label="Download mesh", file_count="single", height=100)
354
  gr.on([btn_stage2.click], infer_stage2, inputs=[text, dropdown, seed, global_info], outputs=[gallery, download])
355
 
356
+ block.launch(share=True, debug=True)