JunhaoZhuang commited on
Commit
b4c4b2a
·
verified ·
1 Parent(s): e7c2a21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -175,11 +175,11 @@ image_encoder = CLIPVisionModelWithProjection.from_pretrained(os.path.join(model
175
  # model_sketch.eval()
176
 
177
 
178
- global pipeline
179
- global MultiResNetModel
180
- global causal_dit
181
- global controlnet
182
- global cur_style
183
 
184
  @spaces.GPU
185
  def load_ckpt():
@@ -298,7 +298,7 @@ def load_ckpt():
298
  print('loaded pipeline')
299
 
300
  load_ckpt()
301
- cur_style = 'line + shadow'
302
  @spaces.GPU
303
  def change_ckpt(style):
304
  global pipeline
 
175
  # model_sketch.eval()
176
 
177
 
178
+ pipeline = None
179
+ MultiResNetModel = None
180
+ causal_dit = None
181
+ controlnet = None
182
+ cur_style = 'line + shadow'
183
 
184
  @spaces.GPU
185
  def load_ckpt():
 
298
  print('loaded pipeline')
299
 
300
  load_ckpt()
301
+
302
  @spaces.GPU
303
  def change_ckpt(style):
304
  global pipeline