Update README.md
Browse files
README.md
CHANGED
|
@@ -9,7 +9,7 @@ You could check our project page at [🏠PromptGuard HomePage](https://prompt-gu
|
|
| 9 |
|
| 10 |
In the future, we will release our training datasets.
|
| 11 |
|
| 12 |
-
#
|
| 13 |
A simple use case of our model is:
|
| 14 |
```python
|
| 15 |
from diffusers import StableDiffusionPipeline
|
|
@@ -35,3 +35,5 @@ prompt_with_system = origin_prompt + " " + token1 + " " + token2 + ...
|
|
| 35 |
image = pipe(prompt).images[0]
|
| 36 |
image.save("example.png")
|
| 37 |
```
|
|
|
|
|
|
|
|
|
| 9 |
|
| 10 |
In the future, we will release our training datasets.
|
| 11 |
|
| 12 |
+
# Inference
|
| 13 |
A simple use case of our model is:
|
| 14 |
```python
|
| 15 |
from diffusers import StableDiffusionPipeline
|
|
|
|
| 35 |
image = pipe(prompt).images[0]
|
| 36 |
image.save("example.png")
|
| 37 |
```
|
| 38 |
+
|
| 39 |
+
To get a better balance between unsafe content moderation and benign content preservation, we recommend you to load Sexual, Political and Disturbing these three safe embeddings.
|