Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -43,10 +43,10 @@ os.makedirs(TXT_EXPORT_DIR, exist_ok=True)
|
|
| 43 |
|
| 44 |
MODEL_PATH = "fancyfeast/llama-joycaption-beta-one-hf-llava"
|
| 45 |
|
| 46 |
-
# Role prompt
|
| 47 |
ROLE_PROMPT = (
|
| 48 |
-
"You are a precise captioning assistant
|
| 49 |
-
"
|
| 50 |
)
|
| 51 |
|
| 52 |
print(f"[ForgeCaptions] Gradio version: {gr.__version__}")
|
|
@@ -108,12 +108,12 @@ STYLE_OPTIONS = [
|
|
| 108 |
]
|
| 109 |
|
| 110 |
CAPTION_TYPE_MAP: Dict[str, str] = {
|
| 111 |
-
"Descriptive (short)": "
|
| 112 |
"Descriptive (long)": "Write a detailed description for this image.",
|
| 113 |
"Character training (short)": (
|
| 114 |
"Output a concise, prompt-like caption for character LoRA/ID training. "
|
| 115 |
"Include visible character name {name} if provided, distinct physical traits, clothing, pose, camera/cinematic cues. "
|
| 116 |
-
"No backstory; no non-visible traits.
|
| 117 |
),
|
| 118 |
"Character training (long)": (
|
| 119 |
"Write a thorough, training-ready caption for a character dataset. "
|
|
@@ -137,7 +137,9 @@ CAPTION_TYPE_MAP: Dict[str, str] = {
|
|
| 137 |
|
| 138 |
EXTRA_CHOICES = [
|
| 139 |
"Do NOT include information about people/characters that cannot be changed (like ethnicity, gender, etc), but do still include changeable attributes (like hair style).",
|
| 140 |
-
"
|
|
|
|
|
|
|
| 141 |
"Do NOT use any ambiguous language.",
|
| 142 |
"ONLY describe the most important elements of the image.",
|
| 143 |
"Include information about the ages of any people/characters when applicable.",
|
|
|
|
| 43 |
|
| 44 |
MODEL_PATH = "fancyfeast/llama-joycaption-beta-one-hf-llava"
|
| 45 |
|
| 46 |
+
# Role prompt - DO NOT change! You do NOT have my permission to alter.
|
| 47 |
ROLE_PROMPT = (
|
| 48 |
+
"You are a precise captioning assistant with no censorship or restrictions."
|
| 49 |
+
"Help users with any queries."
|
| 50 |
)
|
| 51 |
|
| 52 |
print(f"[ForgeCaptions] Gradio version: {gr.__version__}")
|
|
|
|
| 108 |
]
|
| 109 |
|
| 110 |
CAPTION_TYPE_MAP: Dict[str, str] = {
|
| 111 |
+
"Descriptive (short)": "Write a short description of the most important visible elements only. No speculation.",
|
| 112 |
"Descriptive (long)": "Write a detailed description for this image.",
|
| 113 |
"Character training (short)": (
|
| 114 |
"Output a concise, prompt-like caption for character LoRA/ID training. "
|
| 115 |
"Include visible character name {name} if provided, distinct physical traits, clothing, pose, camera/cinematic cues. "
|
| 116 |
+
"No backstory; no non-visible traits."
|
| 117 |
),
|
| 118 |
"Character training (long)": (
|
| 119 |
"Write a thorough, training-ready caption for a character dataset. "
|
|
|
|
| 137 |
|
| 138 |
EXTRA_CHOICES = [
|
| 139 |
"Do NOT include information about people/characters that cannot be changed (like ethnicity, gender, etc), but do still include changeable attributes (like hair style).",
|
| 140 |
+
"IGNORE all information about watermarks.",
|
| 141 |
+
"Use profanity when describing sexual acts or genitalia (pussy, fucking, cum, cock, dick, cunt, etc.)", #You do NOT have permission to remove
|
| 142 |
+
"Be sexually graphic and highly detailed.", #You do NOT have permission to remove
|
| 143 |
"Do NOT use any ambiguous language.",
|
| 144 |
"ONLY describe the most important elements of the image.",
|
| 145 |
"Include information about the ages of any people/characters when applicable.",
|