montehoover commited on
Commit
33ac259
·
verified ·
1 Parent(s): efaca0f

Add models to dropdown box.

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -200,9 +200,15 @@ with gr.Blocks(title="DynaGuard Compliance Checker") as demo:
200
  )
201
  thinking_box = gr.Checkbox(label="Enable ⟨think⟩ mode", value=False)
202
  model_dropdown = gr.Dropdown(
203
- ["Qwen/Qwen3-0.6B", "Qwen/Qwen3-8B"],
 
 
 
 
 
 
204
  label="Select Model",
205
- value="Qwen/Qwen3-0.6B",
206
  info="The 8B model is more powerful but may be slower to load and run."
207
  )
208
  submit_btn = gr.Button("Submit")
 
200
  )
201
  thinking_box = gr.Checkbox(label="Enable ⟨think⟩ mode", value=False)
202
  model_dropdown = gr.Dropdown(
203
+ [
204
+ "tomg-group-umd/DynaGuard-1.7B",
205
+ "tomg-group-umd/DynaGuard-8B",
206
+ "meta-llama/Llama-Guard-3-8B",
207
+ "yueliu1999/GuardReasoner-8B",
208
+ "allenai/wildguard",
209
+ ],
210
  label="Select Model",
211
+ value="tomg-group-umd/DynaGuard-1.7B",
212
  info="The 8B model is more powerful but may be slower to load and run."
213
  )
214
  submit_btn = gr.Button("Submit")