Update for ZeroGPU

app.py

@@ -8,6 +8,7 @@ import random
 import gradio as gr
 import numpy as np
 import PIL.Image
+import spaces
 import torch
 from diffusers import StableDiffusionAttendAndExcitePipeline, StableDiffusionPipeline
 
@@ -47,6 +48,7 @@ def get_token_table(prompt: str) -> list[tuple[int, str]]:
     return list(enumerate(tokens, start=1))
 
 
+@spaces.GPU
 def run(
     prompt: str,
     indices_to_alter_str: str,
@@ -60,7 +62,6 @@ def run(
         20: 0.8,
     },
     max_iter_to_alter: int = 25,
-    progress=gr.Progress(track_tqdm=True),
 ) -> PIL.Image.Image:
     if num_inference_steps > MAX_INFERENCE_STEPS:
         raise gr.Error(f"Number of steps cannot exceed {MAX_INFERENCE_STEPS}.")

requirements.txt

@@ -1,6 +1,7 @@
 accelerate==0.23.0
 diffusers==0.21.2
 Pillow==10.0.1
+spaces==0.14.0
 torch==2.0.0
 torchvision==0.15.1
 transformers==4.33.2