Spaces:
Runtime error
Runtime error
Commit
·
3961dc5
1
Parent(s):
f5e8315
update
Browse files
app.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
import os
|
| 2 |
import cv2
|
| 3 |
import math
|
| 4 |
-
import spaces
|
| 5 |
import torch
|
| 6 |
import random
|
| 7 |
import numpy as np
|
|
@@ -183,7 +183,7 @@ def apply_style(style_name: str, positive: str, negative: str = "") -> tuple[str
|
|
| 183 |
p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
|
| 184 |
return p.replace("{prompt}", positive), n + ' ' + negative
|
| 185 |
|
| 186 |
-
|
| 187 |
def generate_image(face_image, pose_image, prompt, negative_prompt, style_name, enhance_face_region, num_steps, identitynet_strength_ratio, adapter_strength_ratio, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):
|
| 188 |
|
| 189 |
if face_image is None:
|
|
|
|
| 1 |
import os
|
| 2 |
import cv2
|
| 3 |
import math
|
| 4 |
+
#import spaces
|
| 5 |
import torch
|
| 6 |
import random
|
| 7 |
import numpy as np
|
|
|
|
| 183 |
p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
|
| 184 |
return p.replace("{prompt}", positive), n + ' ' + negative
|
| 185 |
|
| 186 |
+
#@spaces.GPU
|
| 187 |
def generate_image(face_image, pose_image, prompt, negative_prompt, style_name, enhance_face_region, num_steps, identitynet_strength_ratio, adapter_strength_ratio, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):
|
| 188 |
|
| 189 |
if face_image is None:
|