wzuidema commited on
Commit
fb52eb2
1 Parent(s): d0460c0

changed labels

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -267,13 +267,13 @@ def sentiment_explanation_hila(input_text, layer):
267
 
268
  return show_explanation(model, input_ids, attention_mask, start_layer=int(layer))
269
 
270
- layer_slider = gradio.Slider(minimum=0, maximum=12, value=8, step=1, label="Select layer")
271
  hila = gradio.Interface(
272
  fn=sentiment_explanation_hila,
273
  inputs=["text", layer_slider],
274
  outputs="html",
275
  )
276
- layer_slider2 = gradio.Slider(minimum=0, maximum=12, value=0, step=1, label="Select layer")
277
  lig = gradio.Interface(
278
  fn=sentence_sentiment,
279
  inputs=["text", layer_slider2],
 
267
 
268
  return show_explanation(model, input_ids, attention_mask, start_layer=int(layer))
269
 
270
+ layer_slider = gradio.Slider(minimum=0, maximum=12, value=8, step=1, label="Select rollout layer")
271
  hila = gradio.Interface(
272
  fn=sentiment_explanation_hila,
273
  inputs=["text", layer_slider],
274
  outputs="html",
275
  )
276
+ layer_slider2 = gradio.Slider(minimum=0, maximum=12, value=0, step=1, label="Select IG layer")
277
  lig = gradio.Interface(
278
  fn=sentence_sentiment,
279
  inputs=["text", layer_slider2],