fffiloni commited on
Commit
3976f22
·
verified ·
1 Parent(s): f159911

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -16,6 +16,7 @@ from diffusers.utils import load_image
16
  # load pipeline
17
  global pipe_canny
18
  global pipe_tile
 
19
  controlnet_canny = SD3ControlNetModel.from_pretrained("InstantX/SD3-Controlnet-Canny")
20
  controlnet_tile = SD3ControlNetModel.from_pretrained("InstantX/SD3-Controlnet-Tile")
21
 
@@ -57,7 +58,7 @@ def infer(image_in, prompt, control_type, inference_steps, guidance_scale, contr
57
  n_prompt = 'NSFW, nude, naked, porn, ugly'
58
 
59
  if control_type == "canny":
60
- global pipe = pipe_canny
61
  pipe.to("cuda", torch.float16)
62
  # Canny preprocessing
63
  image_to_canny = load_image(image_in)
@@ -70,7 +71,7 @@ def infer(image_in, prompt, control_type, inference_steps, guidance_scale, contr
70
  control_image = image_to_canny
71
 
72
  elif control_type == "tile":
73
- global pipe = pipe_tile
74
  pipe.to("cuda", torch.float16)
75
  control_image = load_image(image_in)
76
 
 
16
  # load pipeline
17
  global pipe_canny
18
  global pipe_tile
19
+ global pipe
20
  controlnet_canny = SD3ControlNetModel.from_pretrained("InstantX/SD3-Controlnet-Canny")
21
  controlnet_tile = SD3ControlNetModel.from_pretrained("InstantX/SD3-Controlnet-Tile")
22
 
 
58
  n_prompt = 'NSFW, nude, naked, porn, ugly'
59
 
60
  if control_type == "canny":
61
+ pipe = pipe_canny
62
  pipe.to("cuda", torch.float16)
63
  # Canny preprocessing
64
  image_to_canny = load_image(image_in)
 
71
  control_image = image_to_canny
72
 
73
  elif control_type == "tile":
74
+ pipe = pipe_tile
75
  pipe.to("cuda", torch.float16)
76
  control_image = load_image(image_in)
77