valhalla commited on
Commit
5d133e3
1 Parent(s): 84fcd8e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import diffusers
6
  from share_btn import community_icon_html, loading_icon_html, share_js
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
- pipe = AutoPipelineForInpainting.from_pretrained("diffusers/inpainting-sdxl-0.1", torch_dtype=torch.float16, variant="fp16").to(device)
10
 
11
  def read_content(file_path: str) -> str:
12
  """read the content of target file
 
6
  from share_btn import community_icon_html, loading_icon_html, share_js
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
+ pipe = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype=torch.float16, variant="fp16").to(device)
10
 
11
  def read_content(file_path: str) -> str:
12
  """read the content of target file