Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,7 +26,7 @@ pipe.to(dtype=torch.float16, device=device)
|
|
| 26 |
MAX_SEED = np.iinfo(np.int32).max
|
| 27 |
MAX_IMAGE_SIZE = 1024
|
| 28 |
|
| 29 |
-
@spaces.GPU(
|
| 30 |
def infer(prompt, seed, randomize_seed, tau,
|
| 31 |
guidance_scale):
|
| 32 |
|
|
@@ -98,6 +98,9 @@ with gr.Blocks(css=css) as demo:
|
|
| 98 |
Currently running on {power_device}.
|
| 99 |
"""
|
| 100 |
)
|
|
|
|
|
|
|
|
|
|
| 101 |
gr.Markdown(
|
| 102 |
"If you enjoy the space, feel free to give a β to the <a href='https://github.com/yandex-research/invertible-cd' target='_blank'>Github Repo</a>. [](https://github.com/yandex-research/invertible-cd)"
|
| 103 |
)
|
|
|
|
| 26 |
MAX_SEED = np.iinfo(np.int32).max
|
| 27 |
MAX_IMAGE_SIZE = 1024
|
| 28 |
|
| 29 |
+
@spaces.GPU()
|
| 30 |
def infer(prompt, seed, randomize_seed, tau,
|
| 31 |
guidance_scale):
|
| 32 |
|
|
|
|
| 98 |
Currently running on {power_device}.
|
| 99 |
"""
|
| 100 |
)
|
| 101 |
+
gr.Markdown(
|
| 102 |
+
"Feel free to check out our [image editing demo](https://huggingface.co/spaces/dbaranchuk/iCD-image-editing) as well."
|
| 103 |
+
)
|
| 104 |
gr.Markdown(
|
| 105 |
"If you enjoy the space, feel free to give a β to the <a href='https://github.com/yandex-research/invertible-cd' target='_blank'>Github Repo</a>. [](https://github.com/yandex-research/invertible-cd)"
|
| 106 |
)
|