bug fix

gradio_app/custom_models/mvimg_prediction.py

@@ -15,6 +15,7 @@ checkpoint_path = "ckpt/img2mvimg/unet_state_dict.pth"
 trainer, pipeline = load_pipeline(training_config, checkpoint_path)
 
 def predict(img_list: List[Image.Image], guidance_scale=2., **kwargs):
+    global pipeline
     pipeline = pipeline.to("cuda")
     if isinstance(img_list, Image.Image):
         img_list = [img_list]

gradio_app/custom_models/normal_prediction.py

@@ -10,6 +10,7 @@ checkpoint_path = "ckpt/image2normal/unet_state_dict.pth"
 trainer, pipeline = load_pipeline(training_config, checkpoint_path)
 
 def predict_normals(image: List[Image.Image], guidance_scale=2., do_rotate=True, num_inference_steps=30, **kwargs):    
+    global pipeline
     pipeline = pipeline.to("cuda")
     
     img_list = image if isinstance(image, list) else [image]