Update app.py

app.py

@@ -1,3 +1,15 @@
-import gradio as gr
+from diffusers import DiffusionPipeline
 
-gr.load("models/stabilityai/stable-diffusion-xl-base-1.0").launch()
+model_id = "runwayml/stable-diffusion-v1-5"
+pipeline = DiffusionPipeline.from_pretrained(model_id)
+
+prompt = "portrait photo of a old warrior chief"
+
+pipeline = pipeline.to("cuda")
+
+import torch
+
+generator = torch.Generator("cuda").manual_seed(0)
+
+image = pipeline(prompt, generator=generator).images[0]
+image