Add space

gradio_app.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import gradio as gr
+import spaces
 from PIL import Image
 import torch
 
@@ -22,12 +23,13 @@ scheduler_type = Scheduler_Type.EULER
 image_size = model_type_to_size(Model_Type.SDXL_Turbo)
 pipe_inversion, pipe_inference = get_pipes(model_type, scheduler_type, device=device)
 
-cache_size = 10
+cache_size = 25
 prev_configs = [None for i in range(cache_size)]
 prev_inv_latents = [None for i in range(cache_size)]
 prev_images = [None for i in range(cache_size)]
 prev_noises = [None for i in range(cache_size)]
 
+@spaces.GPU
 def main_pipeline(
         input_image: str,
         src_prompt: str,

requirements.txt

@@ -1,5 +1,6 @@
 gradio
-torch==2.2.1
+spaces
+torch==2.2.0
 torchvision==0.17.1
 diffusers==0.24.0
 transformers==4.32.1