p1atdev commited on
Commit
284b24d
1 Parent(s): 6ba5764

chore: update info

Browse files
Files changed (1) hide show
  1. v2.py +6 -2
v2.py CHANGED
@@ -30,7 +30,7 @@ from utils import ASPECT_RATIO_OPTIONS, RATING_OPTIONS, LENGTH_OPTIONS, IDENTITY
30
  HF_TOKEN = os.getenv("HF_TOKEN", None)
31
 
32
  ALL_MODELS = {
33
- "dart-v2-mixtral-160m-sft-8": {
34
  "repo": "p1atdev/dart-v2-mixtral-160m-sft-8",
35
  "type": "sft",
36
  "class": MixtralModel,
@@ -180,16 +180,19 @@ class V2UI:
180
  )
181
  input_aspect_ratio = gr.Radio(
182
  label="Aspect ratio",
 
183
  choices=["ultra_wide", "wide", "square", "tall", "ultra_tall"],
184
  value="tall",
185
  )
186
  input_length = gr.Radio(
187
  label="Length",
 
188
  choices=list(LENGTH_OPTIONS.keys()),
189
  value="long",
190
  )
191
  input_identity = gr.Radio(
192
- label="Keep identity level",
 
193
  choices=list(IDENTITY_OPTIONS.keys()),
194
  value="none",
195
  )
@@ -197,6 +200,7 @@ class V2UI:
197
  with gr.Accordion(label="Advanced options", open=False):
198
  input_ban_tags = gr.Textbox(
199
  label="Ban tags",
 
200
  placeholder="alternate costumen, ...",
201
  )
202
 
 
30
  HF_TOKEN = os.getenv("HF_TOKEN", None)
31
 
32
  ALL_MODELS = {
33
+ "dart-v2-mixtral-160m-sft": {
34
  "repo": "p1atdev/dart-v2-mixtral-160m-sft-8",
35
  "type": "sft",
36
  "class": MixtralModel,
 
180
  )
181
  input_aspect_ratio = gr.Radio(
182
  label="Aspect ratio",
183
+ info="The aspect ratio of the image.",
184
  choices=["ultra_wide", "wide", "square", "tall", "ultra_tall"],
185
  value="tall",
186
  )
187
  input_length = gr.Radio(
188
  label="Length",
189
+ info="How long the total tags should be.",
190
  choices=list(LENGTH_OPTIONS.keys()),
191
  value="long",
192
  )
193
  input_identity = gr.Radio(
194
+ label="Keep identity",
195
+ info="How strict to keep the identity of the character or subject.",
196
  choices=list(IDENTITY_OPTIONS.keys()),
197
  value="none",
198
  )
 
200
  with gr.Accordion(label="Advanced options", open=False):
201
  input_ban_tags = gr.Textbox(
202
  label="Ban tags",
203
+ info="Tags to ban from the output.",
204
  placeholder="alternate costumen, ...",
205
  )
206