QinOwen commited on
Commit
d4a2e0d
1 Parent(s): f4723f0

remove-base-model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -193,7 +193,7 @@ with gr.Blocks(css=custom_css) as demo:
193
  with gr.Column(elem_id="params"):
194
  lora_model = gr.Dropdown(
195
  label="VADER Model",
196
- choices=["huggingface-pickscore", "huggingface-hps-aesthetic", "Base Model"],
197
  value="huggingface-pickscore"
198
  )
199
  lora_rank = gr.Slider(minimum=8, maximum=16, label="LoRA Rank", step = 8, value=16)
 
193
  with gr.Column(elem_id="params"):
194
  lora_model = gr.Dropdown(
195
  label="VADER Model",
196
+ choices=["huggingface-pickscore", "huggingface-hps-aesthetic"],
197
  value="huggingface-pickscore"
198
  )
199
  lora_rank = gr.Slider(minimum=8, maximum=16, label="LoRA Rank", step = 8, value=16)