# 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("")
# prompt = "retro serie of different cars with different colors and shapes, mdjrny-v4 style"
# image = pipe(prompt).images[0]
# print(image)

from diffusers import StableDiffusionOnnxPipeline

pipe = StableDiffusionOnnxPipeline.from_pretrained(
    "runwayml/stable-diffusion-v1-5",
    revision="onnx",
    provider="CUDAExecutionProvider",
)

prompt = "a photo of an astronaut riding a horse on mars"
image = pipe(prompt).images[0]
print(image)