Update app.py
Browse files
app.py
CHANGED
|
@@ -1,6 +1,8 @@
|
|
| 1 |
import sys
|
| 2 |
sys.path.append('../')
|
| 3 |
|
|
|
|
|
|
|
| 4 |
import torch
|
| 5 |
import random
|
| 6 |
import numpy as np
|
|
@@ -168,6 +170,7 @@ def run_for_examples(source_image, prompt, scale, style_mode):
|
|
| 168 |
style_mode=style_mode,
|
| 169 |
)
|
| 170 |
|
|
|
|
| 171 |
def create_image(input_image,
|
| 172 |
prompt,
|
| 173 |
scale,
|
|
|
|
| 1 |
import sys
|
| 2 |
sys.path.append('../')
|
| 3 |
|
| 4 |
+
import spaces
|
| 5 |
+
|
| 6 |
import torch
|
| 7 |
import random
|
| 8 |
import numpy as np
|
|
|
|
| 170 |
style_mode=style_mode,
|
| 171 |
)
|
| 172 |
|
| 173 |
+
@spaces.GPU
|
| 174 |
def create_image(input_image,
|
| 175 |
prompt,
|
| 176 |
scale,
|