Update app.py

app.py

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