Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 from diffusers import StableDiffusionImg2ImgPipeline
 import torch
@@ -12,6 +13,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = pipe.to(device)
 
 # Define the inference function
+@spaces.GPU
 def ghibli_transform(input_image, prompt="ghibli style", strength=0.75, guidance_scale=7.5, num_steps=50):
     if input_image is None:
         raise gr.Error("No image uploaded! Please upload an image before clicking Transform.")