Spaces:

yangheng
/

Super-Resolution-Anime-Diffusion

Running on Zero

yangheng commited on Aug 25

Commit

1915228

verified ·

1 Parent(s): f80e55b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -40,6 +40,7 @@ import PIL.Image
 import autocuda
 import findfile
 from diffusers import (
     AutoencoderKL,
     UNet2DConditionModel,
@@ -290,6 +291,7 @@ def on_model_change(model_name: str) -> Tuple[gr.update, gr.update]:
         gr.update(placeholder=prefix_text),
     )
 def generate_image(
     mode: str,
     model_name: str,

 import autocuda
 import findfile
+import spaces
 from diffusers import (
     AutoencoderKL,
     UNet2DConditionModel,
         gr.update(placeholder=prefix_text),
     )
+@spaces.GPU(duration=60)
 def generate_image(
     mode: str,
     model_name: str,