Spaces:
Runtime error
Runtime error
@25
Browse files
app.py
CHANGED
|
@@ -2,7 +2,6 @@ import cv2
|
|
| 2 |
import torch
|
| 3 |
import random
|
| 4 |
import numpy as np
|
| 5 |
-
|
| 6 |
import spaces
|
| 7 |
|
| 8 |
import PIL
|
|
@@ -284,7 +283,7 @@ def apply_style(
|
|
| 284 |
p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
|
| 285 |
return p.replace("{prompt}", positive), n + " " + negative
|
| 286 |
|
| 287 |
-
@spaces.GPU(duration=
|
| 288 |
def generate_image(
|
| 289 |
face_image_path,
|
| 290 |
pose_image_path,
|
|
|
|
| 2 |
import torch
|
| 3 |
import random
|
| 4 |
import numpy as np
|
|
|
|
| 5 |
import spaces
|
| 6 |
|
| 7 |
import PIL
|
|
|
|
| 283 |
p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
|
| 284 |
return p.replace("{prompt}", positive), n + " " + negative
|
| 285 |
|
| 286 |
+
@spaces.GPU(duration=25, progress=gr.Progress(track_tqdm=True))
|
| 287 |
def generate_image(
|
| 288 |
face_image_path,
|
| 289 |
pose_image_path,
|