Spaces:
Paused
Paused
Commit
·
4a9ffc4
1
Parent(s):
50550c9
add spaces
Browse files
app.py
CHANGED
|
@@ -4,6 +4,7 @@ import json
|
|
| 4 |
import itertools
|
| 5 |
import math
|
| 6 |
import os
|
|
|
|
| 7 |
import time
|
| 8 |
from pathlib import Path
|
| 9 |
|
|
@@ -397,6 +398,7 @@ def prepare_infer_func():
|
|
| 397 |
|
| 398 |
return image_edit.generate_image
|
| 399 |
|
|
|
|
| 400 |
def inference(prompt, ref_images, seed, size_level, infer_func=None):
|
| 401 |
start_time = time.time()
|
| 402 |
|
|
|
|
| 4 |
import itertools
|
| 5 |
import math
|
| 6 |
import os
|
| 7 |
+
import spaces
|
| 8 |
import time
|
| 9 |
from pathlib import Path
|
| 10 |
|
|
|
|
| 398 |
|
| 399 |
return image_edit.generate_image
|
| 400 |
|
| 401 |
+
@spaces.GPU
|
| 402 |
def inference(prompt, ref_images, seed, size_level, infer_func=None):
|
| 403 |
start_time = time.time()
|
| 404 |
|