tsqn commited on
Commit
dc39771
·
verified ·
1 Parent(s): 745c135

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,7 +20,7 @@ from style_template import styles
20
  from aspect_ratio_template import aspect_ratios
21
 
22
  # global variable
23
- base_model_path = 'SG161222/RealVisXL_V4.0'
24
  face_detector = FaceAnalysis2(providers=['CPUExecutionProvider', 'CUDAExecutionProvider'], allowed_modules=['detection', 'recognition'])
25
  face_detector.prepare(ctx_id=0, det_size=(640, 640))
26
 
@@ -79,7 +79,7 @@ pipe.fuse_lora()
79
  pipe.to(device)
80
 
81
 
82
- @spaces.GPU()
83
  def generate_image(
84
  upload_images,
85
  prompt,
 
20
  from aspect_ratio_template import aspect_ratios
21
 
22
  # global variable
23
+ base_model_path = 'SG161222/RealVisXL_V5.0'
24
  face_detector = FaceAnalysis2(providers=['CPUExecutionProvider', 'CUDAExecutionProvider'], allowed_modules=['detection', 'recognition'])
25
  face_detector.prepare(ctx_id=0, det_size=(640, 640))
26
 
 
79
  pipe.to(device)
80
 
81
 
82
+ @spaces.GPU(duration=120)
83
  def generate_image(
84
  upload_images,
85
  prompt,