Update app.py

app.py

@@ -9,7 +9,7 @@ from auffusion_pipeline import AuffusionPipeline
 
 # ——
 
-from diffusers import StableDiffusionImg2ImgPipeline
+from diffusers import StableDiffusionImg2ImgPipeline, StableDiffusionInpaintPipeline
 from converter import load_wav, mel_spectrogram, normalize_spectrogram, denormalize_spectrogram, Generator, get_mel_spectrogram_from_audio
 from utils import pad_spec, image_add_color, torch_to_pil, normalize, denormalize, prepare_mask_and_masked_image
 
@@ -139,7 +139,7 @@ def infer_inp(prompt, audio_path, progress=gr.Progress(track_tqdm=True)):
     vocoder = Generator.from_pretrained(pretrained_model_name_or_path, subfolder="vocoder")
     vocoder = vocoder.to(device=device, dtype=dtype)
 
-    pipe = StableDiffusionImg2ImgPipeline.from_pretrained(pretrained_model_name_or_path, torch_dtype=dtype)
+    pipe = StableDiffusionInpaintPipeline.from_pretrained(pretrained_model_name_or_path, torch_dtype=dtype)
     pipe = pipe.to(device)
 
     width_start, width = 256, 512