lvyb commited on
Commit
fe1d3d5
β€’
1 Parent(s): 15becb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -14,7 +14,11 @@ examples = [[os.path.join(os.path.dirname(__file__), "lion.jpg")], [os.path.join
14
  app = gr.Interface.from_pipeline(pipe, examples=examples, title='ε›Ύη‰‡θ―†εˆ«')
15
 
16
 
17
- pipe = pipeline("dreamlike-art/dreamlike-photoreal-2.0")
 
 
 
 
18
  examples = [["落ζ—₯"], ["ζ²™ζ»©"]]
19
  app_v1 = gr.Interface.from_pipeline(pipe, examples=examples, title='ε›Ύη‰‡η”Ÿζˆ')
20
 
 
14
  app = gr.Interface.from_pipeline(pipe, examples=examples, title='ε›Ύη‰‡θ―†εˆ«')
15
 
16
 
17
+ from diffusers import StableDiffusionPipeline
18
+ import torch
19
+
20
+ model_id = "dreamlike-art/dreamlike-photoreal-2.0"
21
+ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
22
  examples = [["落ζ—₯"], ["ζ²™ζ»©"]]
23
  app_v1 = gr.Interface.from_pipeline(pipe, examples=examples, title='ε›Ύη‰‡η”Ÿζˆ')
24