tommy24 commited on
Commit
499ee1a
·
1 Parent(s): d93853b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -1,12 +1,12 @@
1
- # import gradio as gr
2
 
3
- # gr.Interface.load("models/prompthero/openjourney").launch()
4
 
5
- from diffusers import StableDiffusionPipeline
6
- import torch
7
- model_id = "prompthero/openjourney"
8
- pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
9
- pipe = pipe.to("cuda")
10
- prompt = "retro serie of different cars with different colors and shapes, mdjrny-v4 style"
11
- image = pipe(prompt).images[0]
12
- print(image)
 
1
+ import gradio as gr
2
 
3
+ gr.Interface.load("models/prompthero/openjourney").launch()
4
 
5
+ # from diffusers import StableDiffusionPipeline
6
+ # import torch
7
+ # model_id = "prompthero/openjourney"
8
+ # pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
9
+ # pipe = pipe.to("cuda")
10
+ # prompt = "retro serie of different cars with different colors and shapes, mdjrny-v4 style"
11
+ # image = pipe(prompt).images[0]
12
+ # print(image)