ttoosi commited on
Commit
08c2dde
·
1 Parent(s): 8a7ee57

Hide epsilon and simplify hierarchy options

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -47,7 +47,7 @@ examples = [
47
  "method": "Prior-Guided Drift Diffusion",
48
  "reverse_diff": {
49
  "model": "resnet50_robust",
50
- "layer": "layer4",
51
  "initial_noise": 0.5,
52
  "diffusion_noise": 0.002,
53
  "step_size": 0.1,
@@ -439,7 +439,7 @@ with gr.Blocks(title="Human Hallucination Prediction", css="""
439
  inference_type = gr.State(value="Prior-Guided Drift Diffusion")
440
 
441
  with gr.Row():
442
- eps_slider = gr.Slider(minimum=0.0, maximum=40.0, value=40.0, step=0.01, label="Epsilon (Stimulus Fidelity)")
443
  iterations_slider = gr.Slider(minimum=1, maximum=600, value=500, step=1, label="Number of Iterations")
444
 
445
  # Keep these internal defaults fixed (not user-editable in simple UI)
@@ -450,7 +450,7 @@ with gr.Blocks(title="Human Hallucination Prediction", css="""
450
  step_size_slider = gr.Slider(minimum=0.01, maximum=2.0, value=1.0, step=0.01,
451
  label="Prior strength")
452
  layer_choice = gr.Dropdown(
453
- choices=["all", "conv1", "bn1", "relu", "maxpool", "layer1", "layer2", "layer3", "layer4", "avgpool"],
454
  value="all",
455
  label="Hierarchy Level"
456
  )
 
47
  "method": "Prior-Guided Drift Diffusion",
48
  "reverse_diff": {
49
  "model": "resnet50_robust",
50
+ "layer": "layer3",
51
  "initial_noise": 0.5,
52
  "diffusion_noise": 0.002,
53
  "step_size": 0.1,
 
439
  inference_type = gr.State(value="Prior-Guided Drift Diffusion")
440
 
441
  with gr.Row():
442
+ eps_slider = gr.State(value=40.0) # fixed epsilon; hidden in simple UI
443
  iterations_slider = gr.Slider(minimum=1, maximum=600, value=500, step=1, label="Number of Iterations")
444
 
445
  # Keep these internal defaults fixed (not user-editable in simple UI)
 
450
  step_size_slider = gr.Slider(minimum=0.01, maximum=2.0, value=1.0, step=0.01,
451
  label="Prior strength")
452
  layer_choice = gr.Dropdown(
453
+ choices=[("Higher", "all"), ("Intermediate", "layer3"), ("Lower", "layer1")],
454
  value="all",
455
  label="Hierarchy Level"
456
  )