p1atdev commited on
Commit
5928cde
1 Parent(s): 48cea3d

chore: added description and examples

Browse files
Files changed (1) hide show
  1. app.py +45 -0
app.py CHANGED
@@ -129,6 +129,14 @@ def image_generation_config_ui():
129
  ]
130
 
131
 
 
 
 
 
 
 
 
 
132
  def main():
133
 
134
  v2 = V2UI()
@@ -138,6 +146,8 @@ def main():
138
  print("Loaded.")
139
 
140
  with gr.Blocks() as ui:
 
 
141
  with gr.Row():
142
  with gr.Column():
143
  v2.ui()
@@ -163,6 +173,41 @@ def main():
163
  visible=True,
164
  )
165
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
166
  v2.get_generate_btn().click(
167
  parse_upsampling_output(v2.on_generate, image_generator.generate),
168
  inputs=[
 
129
  ]
130
 
131
 
132
+ def description_ui():
133
+ gr.Markdown(
134
+ """
135
+ # Danbooru Tags Transformer V2 Demo
136
+ """
137
+ )
138
+
139
+
140
  def main():
141
 
142
  v2 = V2UI()
 
146
  print("Loaded.")
147
 
148
  with gr.Blocks() as ui:
149
+ description_ui()
150
+
151
  with gr.Row():
152
  with gr.Column():
153
  v2.ui()
 
173
  visible=True,
174
  )
175
 
176
+ gr.Examples(
177
+ examples=[
178
+ ["original", "", "1girl, solo", "832x1216"],
179
+ ["original", "", "3girls, 2boys", "1536x640"],
180
+ [
181
+ "sousou no frieren",
182
+ "frieren (sousou no frieren)",
183
+ "1girl, solo",
184
+ "832x1216",
185
+ ],
186
+ [
187
+ "honkai: star rail",
188
+ "silver wolf (honkai: star rail)",
189
+ "1girl, solo",
190
+ "832x1216",
191
+ ],
192
+ [
193
+ "bocchi the rock!",
194
+ "gotoh hitori, kita ikuyo, ijichi nijika, yamada ryo",
195
+ "4girls, multiple girls",
196
+ "1216x832",
197
+ ],
198
+ [
199
+ "chuunibyou demo koi ga shitai!",
200
+ "takanashi rikka",
201
+ "1girl, solo",
202
+ "640x1536",
203
+ ],
204
+ ],
205
+ inputs=[
206
+ *v2.get_inputs()[1:4],
207
+ image_generation_config_components[0], # image size
208
+ ],
209
+ )
210
+
211
  v2.get_generate_btn().click(
212
  parse_upsampling_output(v2.on_generate, image_generator.generate),
213
  inputs=[