mattyamonaca commited on
Commit
2bc849d
1 Parent(s): bbbffe1
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,7 +11,7 @@ import os
11
  import numpy as np
12
  from PIL import Image
13
  import zipfile
14
- #import spaces
15
 
16
  path = os.getcwd()
17
  output_dir = f"{path}/output"
@@ -23,7 +23,7 @@ load_cn_config(cn_lineart_dir)
23
  pipe = get_cn_pipeline()
24
 
25
 
26
- #@spaces.GPU(duration=120)
27
  def generate(detectors, prompt, negative_prompt, reference_flg=False, reference_img=None):
28
  default_pos = ""
29
  default_neg = ""
 
11
  import numpy as np
12
  from PIL import Image
13
  import zipfile
14
+ import spaces
15
 
16
  path = os.getcwd()
17
  output_dir = f"{path}/output"
 
23
  pipe = get_cn_pipeline()
24
 
25
 
26
+ @spaces.GPU(duration=120)
27
  def generate(detectors, prompt, negative_prompt, reference_flg=False, reference_img=None):
28
  default_pos = ""
29
  default_neg = ""