gokaygokay commited on
Commit
eef1455
1 Parent(s): ac150a8

prompt_types

Browse files
Files changed (1) hide show
  1. ui_components.py +10 -3
ui_components.py CHANGED
@@ -101,9 +101,16 @@ def create_interface():
101
  (You need to use Generate Prompt first)""", open=False):
102
  happy_talk = gr.Checkbox(label="Happy Talk", value=True)
103
  compress = gr.Checkbox(label="Compress", value=True)
104
- compression_level = gr.Radio(["soft", "medium", "hard"], label="Compression Level", value="hard")
105
- prompt_type = gr.Radio(["happy", "simple", "poster", "only_objects", "no_figure", "landscape", "fantasy"],
106
- label="Prompt Type", value="happy")
 
 
 
 
 
 
 
107
  custom_base_prompt = gr.Textbox(label="Custom Base Prompt", lines=5)
108
  generate_text_button = gr.Button("Generate Prompt with LLM (Llama 3.1 70B)")
109
  text_output = gr.Textbox(label="Generated Text", lines=10)
 
101
  (You need to use Generate Prompt first)""", open=False):
102
  happy_talk = gr.Checkbox(label="Happy Talk", value=True)
103
  compress = gr.Checkbox(label="Compress", value=True)
104
+ compression_level = gr.Dropdown(
105
+ choices=["soft", "medium", "hard"],
106
+ label="Compression Level",
107
+ value="hard"
108
+ )
109
+ prompt_type = gr.Dropdown(
110
+ choices=["happy", "simple", "poster", "only_objects", "no_figure", "landscape", "fantasy"],
111
+ label="Prompt Type",
112
+ value="happy"
113
+ )
114
  custom_base_prompt = gr.Textbox(label="Custom Base Prompt", lines=5)
115
  generate_text_button = gr.Button("Generate Prompt with LLM (Llama 3.1 70B)")
116
  text_output = gr.Textbox(label="Generated Text", lines=10)