import gradio as gr

gr.Interface.load("models/prompthero/openjourney").launch()

# from diffusers import StableDiffusionPipeline
# import torch
# model_id = "prompthero/openjourney"
# pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
# pipe = pipe.to("cuda")
# prompt = "retro serie of different cars with different colors and shapes, mdjrny-v4 style"
# image = pipe(prompt).images[0]
# print(image)