Update app.py

app.py

@@ -5,8 +5,7 @@ from diffusers import AutoPipelineForInpainting, UNet2DConditionModel
 import diffusers
 from share_btn import community_icon_html, loading_icon_html, share_js
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype=torch.float16, variant="fp16").to(device)
+pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V5.0_noVAE", variant="fp16")
 
 def read_content(file_path: str) -> str:
     """read the content of target file
@@ -80,7 +79,7 @@ with image_blocks as demo:
     gr.HTML(read_content("header.html"))
     with gr.Row():
                 with gr.Column():
-                    image = gr.Image(tool='sketch', elem_id="image_upload", type="pil", label="Upload",height=400)
+                    image = gr.Image(elem_id="image_upload", type="pil", label="Upload",height=400,value="sketch")
                     with gr.Row(elem_id="prompt-container", mobile_collapse=False, equal_height=True):
                         with gr.Row():
                             prompt = gr.Textbox(placeholder="Your prompt (what you want in place of what is erased)", show_label=False, elem_id="prompt")