Moibe commited on
Commit
2cac378
·
verified ·
1 Parent(s): 569b0ef
Files changed (1) hide show
  1. app.py +1 -2
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=20, progress=gr.Progress(track_tqdm=True))
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,