mswhite commited on
Commit
6f3e119
·
1 Parent(s): f8303a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ import gradio as gr
16
  import torch
17
  from diffusers import StableDiffusionPipeline
18
 
19
- pipe = StableDiffusionPipeline.from_pretrained("mswhite/miamivice"), torch_dtype=torch.float16)
20
  pipe = pipe.to("cuda")
21
 
22
  #prompt = "a photo of an astronaut riding a horse on mars"
 
16
  import torch
17
  from diffusers import StableDiffusionPipeline
18
 
19
+ pipe = StableDiffusionPipeline.from_pretrained("mswhite/miamivice", torch_dtype=torch.float16)
20
  pipe = pipe.to("cuda")
21
 
22
  #prompt = "a photo of an astronaut riding a horse on mars"