changes for ZeroGPU

app.py

@@ -2,6 +2,7 @@ import diffusers
 import gradio as gr
 import torch
 from diffusers import AutoPipelineForInpainting
+import spaces
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = AutoPipelineForInpainting.from_pretrained(
@@ -18,7 +19,7 @@ def read_content(file_path: str) -> str:
 
     return content
 
-
+@spaces.GPU()
 def predict(
     input_image,
     prompt="",