Sourabh2
/

Human_Face_generator

StableDiffusionPipeline

Model card Files Files and versions

Metrics Training metrics Community

Sourabh2 commited on Aug 11, 2024

Commit

e68f852

·

verified ·

1 Parent(s): a9f9911

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -10,11 +10,11 @@ model_path = "Sourabh2/Human_Face_generator"
 pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
 pipe.to("cuda")
-prompt = ""
 image = pipe(prompt=prompt).images[0]
 image.save("generated_image.png")
 ```
 ## Example Output
-![pareto](https://huggingface.co/Sourabh2/Human_Face_generator/blob/main/bread.png)

 pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
 pipe.to("cuda")
+prompt = "a man with blonde hair and beard"
 image = pipe(prompt=prompt).images[0]
 image.save("generated_image.png")
 ```
 ## Example Output
+![pareto](https://huggingface.co/Sourabh2/Human_Face_generator/resolve/main/bread.png)