changes for ZeroGPU (#44)

- changes for ZeroGPU (99d346d1e8542d67e57622e912f4cf72609f9faa)


Co-authored-by: Linoy Tsaban <linoyts@users.noreply.huggingface.co>

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="",