Add code for SD v1.5 and main method

app.py

@@ -24,4 +24,17 @@ pipe = pipe.to(device)
 torch.backends.cudnn.benchmark = True
 
 # SD v1.5
-pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
+def get_stable_diffusion_v15_pipeline():
+  model_id = "runwayml/stable-diffusion-v1-5"
+  pipeline = DiffusionPipeline.from_pretrained(mode_id)
+  pipeline = pipeline.to(device)
+  return pipeline
+  
+# main
+def main():
+  prompt = "Hello world, I'm vizard,"
+  
+  pipeline = get_stable_diffusion_v15_pipeline()
+  images = pipeline(prompt).images
+
+main