Vignesh455 commited on
Commit
9dae3a4
·
verified ·
1 Parent(s): 016f9e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from diffusers import AutoPipelineForInpainting, UNet2DConditionModel
5
  import diffusers
6
  from share_btn import community_icon_html, loading_icon_html, share_js
7
 
8
- pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V5.0_noVAE")
9
 
10
  def read_content(file_path: str) -> str:
11
  """read the content of target file
 
5
  import diffusers
6
  from share_btn import community_icon_html, loading_icon_html, share_js
7
 
8
+ pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V5.0_noVAE").to("cuda")
9
 
10
  def read_content(file_path: str) -> str:
11
  """read the content of target file