Spaces:
Restarting
Restarting
Update single.py
Browse files
single.py
CHANGED
@@ -4,6 +4,7 @@ from ferret import Benchmark
|
|
4 |
from torch.nn.functional import softmax
|
5 |
|
6 |
DEFAULT_MODEL = "cardiffnlp/twitter-xlm-roberta-base-sentiment"
|
|
|
7 |
|
8 |
|
9 |
@st.cache()
|
@@ -60,7 +61,7 @@ def body():
|
|
60 |
help="Class label you want to explain.",
|
61 |
)
|
62 |
|
63 |
-
text = st.text_input("Text",
|
64 |
compute = st.button("Run")
|
65 |
|
66 |
if compute and model_name:
|
|
|
4 |
from torch.nn.functional import softmax
|
5 |
|
6 |
DEFAULT_MODEL = "cardiffnlp/twitter-xlm-roberta-base-sentiment"
|
7 |
+
DEFAULT_QUERY = "Great movie for a great nap!"
|
8 |
|
9 |
|
10 |
@st.cache()
|
|
|
61 |
help="Class label you want to explain.",
|
62 |
)
|
63 |
|
64 |
+
text = st.text_input("Text", DEFAULT_QUERY)
|
65 |
compute = st.button("Run")
|
66 |
|
67 |
if compute and model_name:
|