Update app.py
Browse files
app.py
CHANGED
@@ -367,8 +367,7 @@ def main(device, segment_type):
|
|
367 |
image = Image.fromarray((image * 255.0).clip(0, 255).astype(np.uint8))
|
368 |
return image
|
369 |
|
370 |
-
|
371 |
-
@spaces.GPU
|
372 |
def generate_image(prompt1, negative_prompt, man, woman, resolution, local_prompt1, local_prompt2, seed, condition, condition_img1, style):
|
373 |
# try:
|
374 |
path1 = lorapath_man[man]
|
|
|
367 |
image = Image.fromarray((image * 255.0).clip(0, 255).astype(np.uint8))
|
368 |
return image
|
369 |
|
370 |
+
@spaces.GPU(duration=200)
|
|
|
371 |
def generate_image(prompt1, negative_prompt, man, woman, resolution, local_prompt1, local_prompt2, seed, condition, condition_img1, style):
|
372 |
# try:
|
373 |
path1 = lorapath_man[man]
|