Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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/
|
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,
|