mr-szgz commited on
Commit
3e2ddba
·
1 Parent(s): f68b47e

removes non-functional models and displays base-model

Browse files
Files changed (1) hide show
  1. app.py +22 -14
app.py CHANGED
@@ -263,16 +263,18 @@ LORA_SPECS: list[LoRASpec] = [
263
  "url": "https://huggingface.co/renderartist/Coloring-Book-Z-Image-Turbo-LoRA/resolve/main/Coloring_Book_Z_Image_Turbo_v1_renderartist_2000.safetensors",
264
  "adapter": "renderartist",
265
  },
266
- {
267
- "name": "Playtime AI",
268
- "url": "PLACEHOLDER_URL_FOR_CIVITAI_MODEL_2184429",
269
- "adapter": "playtime_ai",
270
- },
271
- {
272
- "name": "Foo Samaritan",
273
- "url": "PLACEHOLDER_URL_FOR_CIVITAI_MODEL_2249185",
274
- "adapter": "foo_samaritan",
275
- },
 
 
276
  ]
277
 
278
  NODES: dict[str, LoRAGeneratorNode] = {
@@ -374,14 +376,20 @@ def generate_image(
374
 
375
  with gr.Blocks() as demo:
376
  gr.Markdown("# Coloring Book Generator")
377
- gr.Markdown("Test the generator nodes with different LoRAs")
378
 
379
  with gr.Row():
380
  with gr.Column():
381
- node_selector = gr.Dropdown(
382
- choices=list(NODES.keys()),
 
 
 
 
 
383
  value="Renderartist",
384
- label="Select Generator Node",
 
385
  )
386
 
387
  example_selector = gr.Dropdown(
 
263
  "url": "https://huggingface.co/renderartist/Coloring-Book-Z-Image-Turbo-LoRA/resolve/main/Coloring_Book_Z_Image_Turbo_v1_renderartist_2000.safetensors",
264
  "adapter": "renderartist",
265
  },
266
+ # LEAVE HERE FOR REFERENCE
267
+ # {
268
+ # "name": "Playtime AI",
269
+ # "url": "PLACEHOLDER_URL_FOR_CIVITAI_MODEL_2184429",
270
+ # "adapter": "playtime_ai",
271
+ # },
272
+ # {
273
+ # "name": "Foo Samaritan",
274
+ # "url": "PLACEHOLDER_URL_FOR_CIVITAI_MODEL_2249185",
275
+ # "adapter": "foo_samaritan",
276
+ # },
277
+ # LEAVE HERE FOR REFERENCE
278
  ]
279
 
280
  NODES: dict[str, LoRAGeneratorNode] = {
 
376
 
377
  with gr.Blocks() as demo:
378
  gr.Markdown("# Coloring Book Generator")
379
+ gr.Markdown("Generate coloring book images with the Renderartist LoRA.")
380
 
381
  with gr.Row():
382
  with gr.Column():
383
+ base_model_display = gr.Textbox(
384
+ value=BASE_MODEL_ID,
385
+ label="Selected Base Model",
386
+ interactive=False,
387
+ )
388
+
389
+ node_selector = gr.Textbox(
390
  value="Renderartist",
391
+ label="Selected Model Adapter (LoRA)",
392
+ interactive=False,
393
  )
394
 
395
  example_selector = gr.Dropdown(