wzuidema
commited on
Commit
•
fb52eb2
1
Parent(s):
d0460c0
changed labels
Browse files
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],
|