Spaces:
Runtime error
Runtime error
Yeb Havinga
commited on
Commit
•
5100a8a
1
Parent(s):
7ddfbb5
Update demo
Browse files
app.py
CHANGED
@@ -8,8 +8,8 @@ from transformers import set_seed
|
|
8 |
|
9 |
debug = False
|
10 |
|
11 |
-
MODEL = "flax-community/t5-base-dutch-demo"
|
12 |
-
|
13 |
|
14 |
DEFAULT_TEXT: str = """Bij meerdere dijken in gemeenten langs de Maas in de veiligheidsregio Limburg-Noord zijn lekkages ontstaan. Dat vertelde burgemeester van Venlo Antoin Scholten - tevens voorzitter van de veiligheidsregio - zaterdagmiddag op een persconferentie. Een bedrijventerrein aan de Prins Willem Alexanderhaven in Roermond wordt om die reden ontruimd. Scholten vraagt Nederlanders de komende dagen weg te blijven uit de regio.
|
15 |
"We zien dat zwakke plekken zich voordoen", aldus Scholten. "De situatie is nog onzeker. Het water is ongelofelijk grillig." Volgens hem is het onzeker of de dijken langs de Prins Willem Alexanderhaven het gaan houden. Daar zou al water doorheen sijpelen.
|
@@ -39,6 +39,17 @@ generator_kwargs_top_k = {
|
|
39 |
}
|
40 |
|
41 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
42 |
class TextSummarizer:
|
43 |
def __init__(self):
|
44 |
self.tokenizer = None
|
@@ -81,18 +92,21 @@ def main():
|
|
81 |
|
82 |
generator = instantiate_generator()
|
83 |
|
84 |
-
st.
|
|
|
|
|
|
|
|
|
|
|
85 |
|
86 |
input_text = st.text_area("Enter a Dutch news text", DEFAULT_TEXT)
|
87 |
|
88 |
-
|
89 |
-
|
90 |
-
if button_clicked:
|
91 |
if debug:
|
92 |
response = """Een bedrijventerrein aan de Prins Willem Alexanderhaven in Roermond heeft de komende dagen weg te blijven uit de regio. De burgemeester van Roermond roept alle ramptoeristen er opnieuw toe op meer dan 10.00 uur weg."""
|
93 |
else:
|
94 |
response = generator.summarize(input_text, generator_kwargs_beam)
|
95 |
-
st.header("
|
96 |
st.markdown(response)
|
97 |
|
98 |
|
|
|
8 |
|
9 |
debug = False
|
10 |
|
11 |
+
# MODEL = "flax-community/t5-base-dutch-demo"
|
12 |
+
MODEL: str = "."
|
13 |
|
14 |
DEFAULT_TEXT: str = """Bij meerdere dijken in gemeenten langs de Maas in de veiligheidsregio Limburg-Noord zijn lekkages ontstaan. Dat vertelde burgemeester van Venlo Antoin Scholten - tevens voorzitter van de veiligheidsregio - zaterdagmiddag op een persconferentie. Een bedrijventerrein aan de Prins Willem Alexanderhaven in Roermond wordt om die reden ontruimd. Scholten vraagt Nederlanders de komende dagen weg te blijven uit de regio.
|
15 |
"We zien dat zwakke plekken zich voordoen", aldus Scholten. "De situatie is nog onzeker. Het water is ongelofelijk grillig." Volgens hem is het onzeker of de dijken langs de Prins Willem Alexanderhaven het gaan houden. Daar zou al water doorheen sijpelen.
|
|
|
39 |
}
|
40 |
|
41 |
|
42 |
+
HEADER_INFO = """
|
43 |
+
# Chef Transformer 🍲
|
44 |
+
This demo uses [T5 trained on RecipeNLG](https://huggingface.co/flax-community/t5-recipe-generation)
|
45 |
+
to generate recipe from a given set of ingredients
|
46 |
+
""".strip()
|
47 |
+
SIDEBAR_INFO = """
|
48 |
+
# Configuration
|
49 |
+
""".strip()
|
50 |
+
PROMPT_BOX = "Add custom ingredients here (separated by `,`): "
|
51 |
+
|
52 |
+
|
53 |
class TextSummarizer:
|
54 |
def __init__(self):
|
55 |
self.tokenizer = None
|
|
|
92 |
|
93 |
generator = instantiate_generator()
|
94 |
|
95 |
+
st.markdown(
|
96 |
+
"""# Netherformer 📰
|
97 |
+
This demo uses [T5 Dutch finetuned on CNN DM](https://huggingface.co/flax-community/t5-base-dutch-demo)
|
98 |
+
to generate summaries of a news story
|
99 |
+
"""
|
100 |
+
)
|
101 |
|
102 |
input_text = st.text_area("Enter a Dutch news text", DEFAULT_TEXT)
|
103 |
|
104 |
+
if st.button("Generate summary"):
|
|
|
|
|
105 |
if debug:
|
106 |
response = """Een bedrijventerrein aan de Prins Willem Alexanderhaven in Roermond heeft de komende dagen weg te blijven uit de regio. De burgemeester van Roermond roept alle ramptoeristen er opnieuw toe op meer dan 10.00 uur weg."""
|
107 |
else:
|
108 |
response = generator.summarize(input_text, generator_kwargs_beam)
|
109 |
+
st.header("Summary:")
|
110 |
st.markdown(response)
|
111 |
|
112 |
|
style.css
CHANGED
@@ -5,12 +5,12 @@ body {
|
|
5 |
display: flex;
|
6 |
justify-content: center;
|
7 |
}
|
8 |
-
|
9 |
-
color: #4F8BF9
|
10 |
-
border-radius: 50
|
11 |
-
height: 3em
|
12 |
-
width: 3em
|
13 |
-
}
|
14 |
|
15 |
.stTextInput>div>div>input {
|
16 |
color: #4F8BF9;
|
|
|
5 |
display: flex;
|
6 |
justify-content: center;
|
7 |
}
|
8 |
+
/*.stButton>button {*/
|
9 |
+
/* color: #4F8BF9;*/
|
10 |
+
/* border-radius: 50%;*/
|
11 |
+
/* height: 3em;*/
|
12 |
+
/* width: 3em;*/
|
13 |
+
/*}*/
|
14 |
|
15 |
.stTextInput>div>div>input {
|
16 |
color: #4F8BF9;
|