gokaygokay commited on
Commit
db63048
1 Parent(s): cd63ff7
Files changed (1) hide show
  1. app.py +19 -18
app.py CHANGED
@@ -176,24 +176,25 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft(primary_hue="blue", secondar
176
  text_prompt = gr.Textbox(label="Text Prompt (optional, used if no image is uploaded)")
177
  use_enhancer = gr.Checkbox(label="Use Prompt Enhancer", value=False)
178
  use_llm_generator = gr.Checkbox(label="Use LLM Prompt Generator", value=False)
179
- llm_provider = gr.Dropdown(
180
- choices=["Hugging Face", "SambaNova"],
181
- label="LLM Provider",
182
- value="Hugging Face",
183
- visible=False
184
- )
185
- llm_model = gr.Dropdown(
186
- label="LLM Model",
187
- choices=["Qwen/Qwen2.5-72B-Instruct", "meta-llama/Meta-Llama-3.1-70B-Instruct", "mistralai/Mixtral-8x7B-Instruct-v0.1", "mistralai/Mistral-7B-Instruct-v0.3"],
188
- value="Qwen/Qwen2.5-72B-Instruct",
189
- visible=False
190
- )
191
- prompt_type = gr.Dropdown(
192
- choices=["Random", "Long", "Short", "Medium", "OnlyObjects", "NoFigure", "Landscape", "Fantasy"],
193
- label="Prompt Type",
194
- value="Random",
195
- visible=False
196
- )
 
197
  generate_prompt_btn = gr.Button("Generate Prompt", elem_classes="submit-btn")
198
  final_prompt = gr.Textbox(label="Final Prompt", interactive=False)
199
 
 
176
  text_prompt = gr.Textbox(label="Text Prompt (optional, used if no image is uploaded)")
177
  use_enhancer = gr.Checkbox(label="Use Prompt Enhancer", value=False)
178
  use_llm_generator = gr.Checkbox(label="Use LLM Prompt Generator", value=False)
179
+ with gr.Accordion("Image Settings", open=False):
180
+ llm_provider = gr.Dropdown(
181
+ choices=["Hugging Face", "SambaNova"],
182
+ label="LLM Provider",
183
+ value="Hugging Face",
184
+ visible=False
185
+ )
186
+ llm_model = gr.Dropdown(
187
+ label="LLM Model",
188
+ choices=["Qwen/Qwen2.5-72B-Instruct", "meta-llama/Meta-Llama-3.1-70B-Instruct", "mistralai/Mixtral-8x7B-Instruct-v0.1", "mistralai/Mistral-7B-Instruct-v0.3"],
189
+ value="Qwen/Qwen2.5-72B-Instruct",
190
+ visible=False
191
+ )
192
+ prompt_type = gr.Dropdown(
193
+ choices=["Random", "Long", "Short", "Medium", "OnlyObjects", "NoFigure", "Landscape", "Fantasy"],
194
+ label="Prompt Type",
195
+ value="Random",
196
+ visible=False
197
+ )
198
  generate_prompt_btn = gr.Button("Generate Prompt", elem_classes="submit-btn")
199
  final_prompt = gr.Textbox(label="Final Prompt", interactive=False)
200