Update src/pipelines/inpainting.py

src/pipelines/inpainting.py

@@ -1,13 +1,14 @@
 import torch
+import spaces
 import gradio as gr
 from src.util.base import *
 from src.util.params import *
 from diffusers import AutoPipelineForInpainting
 
-# inpaint_pipe = AutoPipelineForInpainting.from_pretrained(inpaint_model_path).to(torch_device)
-inpaint_pipe = AutoPipelineForInpainting.from_pipe(pipe).to(torch_device)
-
+inpaint_pipe = AutoPipelineForInpainting.from_pretrained(inpaint_model_path).to(torch_device)
+# inpaint_pipe = AutoPipelineForInpainting.from_pipe(pipe).to(torch_device)
 
+@spaces.GPU(enable_queue=True)
 def inpaint(dict, num_inference_steps, seed, prompt="", progress=gr.Progress()):
     progress(0)
     mask = dict["mask"].convert("RGB").resize((imageHeight, imageWidth))