TenPoisk commited on
Commit
4dc6891
1 Parent(s): 3b83fb9

Update midjourney.py

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