Update src/pipelines/interpolate.py

src/pipelines/interpolate.py

@@ -1,4 +1,5 @@
 import torch
+import spaces
 import gradio as gr
 from src.util.base import *
 from src.util.params import *
@@ -17,7 +18,7 @@ def interpolate_prompts(promptA, promptB, num_interpolation_steps):
 
     return interpolated_embeddings
 
-
+@spaces.GPU(enable_queue=True)
 def display_interpolate_images(
     seed, promptA, promptB, num_inference_steps, num_images, progress=gr.Progress()
 ):