jerpint commited on
Commit
e4e8b82
·
1 Parent(s): 99345a3

instantiate directly

Browse files
Files changed (1) hide show
  1. app.py +1 -6
app.py CHANGED
@@ -10,11 +10,6 @@ from game_of_life import GameOfLife
10
  from utils import resize_image, generate_image_from_grid
11
 
12
 
13
- @spaces.GPU
14
- def init_controlnet(device: str) -> QRControlNet:
15
- return QRControlNet(device=device)
16
-
17
-
18
  def generate_all_images(
19
  gol_grids: list[np.array],
20
  source_image: Image,
@@ -115,7 +110,7 @@ device = "cuda"
115
  # device = "mps"
116
  # device = "cpu"
117
  print(f"Using {device=}")
118
- controlnet = init_controlnet(device=device)
119
 
120
 
121
  source_image = gr.Image(label="Source Image", type="pil", value="sky-gol-image.jpeg")
 
10
  from utils import resize_image, generate_image_from_grid
11
 
12
 
 
 
 
 
 
13
  def generate_all_images(
14
  gol_grids: list[np.array],
15
  source_image: Image,
 
110
  # device = "mps"
111
  # device = "cpu"
112
  print(f"Using {device=}")
113
+ controlnet = QRControlNet(device=device)
114
 
115
 
116
  source_image = gr.Image(label="Source Image", type="pil", value="sky-gol-image.jpeg")