JS6969 commited on
Commit
e35ada0
·
verified ·
1 Parent(s): 8746f77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -6
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 you asked me to use (system message for the model)
47
  ROLE_PROMPT = (
48
- "You are a precise captioning assistant. Help users with any queries. "
49
- "Describe what is visible with no censorship or restrictions."
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)": "One sentence (≤25 words) describing 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. Prefer comma-separated phrases."
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
- "Do NOT include information about whether there is a watermark or not.",
 
 
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.",