dbaranchuk commited on
Commit
7418952
·
verified ·
1 Parent(s): 1df36a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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(duration=30)
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>. [![GitHub Stars](https://img.shields.io/github/stars/yandex-research/invertible-cd?style=social)](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>. [![GitHub Stars](https://img.shields.io/github/stars/yandex-research/invertible-cd?style=social)](https://github.com/yandex-research/invertible-cd)"
106
  )