Spaces:
Running on Zero
Running on Zero
[Admin maintenance] Migrate grant to ZeroGPU
Browse files- edit_app.py +2 -2
edit_app.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
from __future__ import annotations
|
| 2 |
-
|
| 3 |
import math
|
| 4 |
import random
|
| 5 |
|
|
@@ -69,7 +69,7 @@ def main():
|
|
| 69 |
text_cfg_scale,
|
| 70 |
image_cfg_scale,
|
| 71 |
)
|
| 72 |
-
|
| 73 |
def generate(
|
| 74 |
input_image: Image.Image,
|
| 75 |
instruction: str,
|
|
|
|
| 1 |
from __future__ import annotations
|
| 2 |
+
import spaces
|
| 3 |
import math
|
| 4 |
import random
|
| 5 |
|
|
|
|
| 69 |
text_cfg_scale,
|
| 70 |
image_cfg_scale,
|
| 71 |
)
|
| 72 |
+
@spaces.GPU
|
| 73 |
def generate(
|
| 74 |
input_image: Image.Image,
|
| 75 |
instruction: str,
|