Update app.py

app.py

@@ -2,11 +2,23 @@
 
 # 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"
+# 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)