upload

app.py

@@ -6,8 +6,8 @@ from PIL import Image
 from tqdm import tqdm
 from train.scripts.generate_lm_multiple import gen_sequence, build_visorgpt
 from utils.seq2coord import gen_cond_mask
-from gligen.gligen_inference_box import gligen_infer, build_gligen_model
-from controlnet.gradio_pose2image_v2 import control_infer, build_control_model, build_controlv11_model
+from visor_gligen.gligen_inference_box import gligen_infer, build_gligen_model
+from visor_controlnet.gradio_pose2image_v2 import control_infer, build_control_model, build_controlv11_model
 
 # init models 
 visorgpt_config_path = 'train/models/gpt2/config.json'