Hugging Face's logo

Spaces:
Mlaure94
/
Paint-coiffure
Running

App Files Community
Mlaure94 commited on
Commit
bbbc6f9
·
verified ·
1 Parent(s): 44234d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ pipe = DiffusionPipeline.from_pretrained(
21
  "Fantasy-Studio/Paint-by-Example",
22
  torch_dtype=torch.float16,
23
  )
24
- pipe = pipe.to("device")
25
 
26
  from share_btn import community_icon_html, loading_icon_html, share_js
27
 
 
21
  "Fantasy-Studio/Paint-by-Example",
22
  torch_dtype=torch.float16,
23
  )
24
+ pipe = pipe.to(device)
25
 
26
  from share_btn import community_icon_html, loading_icon_html, share_js
27