Yeb Havinga commited on
Commit
71637c8
1 Parent(s): c013713

Add t5-large model. Show model-load spinner

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -11,7 +11,7 @@ debug = False
11
  MODELS = [
12
  "flax-community/t5-base-dutch-demo",
13
  "yhavinga/t5-v1.1-base-dutch-cnn-test",
14
- # "yhavinga/t5-v1.1-large-dutch-cnn-test",
15
  ]
16
 
17
  DEFAULT_TEXT: str = """(CNN) Skywatchers in West-Noord-Amerika zijn in voor een traktatie: een bijna vijf minuten totale maansverduistering vanmorgen. Hier is hoe het zich ontvouwt:. Het begon om 3:16 a.m. Pacific Daylight Tijd, toen de maan begon te bewegen in de schaduw van de Aarde. Voor het volgende uur en 45 minuten, die schaduw zal bewegen over de maan en verzwolgen het om 4:58 a.m. Pacific Time. De totale verduistering zal slechts vier minuten en 43 seconden duren, en NASA zegt dat maakt het de kortste van de eeuw. Kijken live op NASA TV. Terwijl mensen ten westen van de Mississippi River zal het beste uitzicht hebben, ten minste een gedeeltelijke verduistering zal zichtbaar zijn over de hele natie. Maar zonsopgang zal de show te onderbreken op de Oostkust. Delen van Zuid-Amerika, India, China en China Een maansverduistering gebeurt wanneer de zon, de aarde en de maan een rechte lijn vormen in de ruimte, met de aarde in het midden. De zon schijnt op de Aarde en creëert een schaduw. Als de maan dieper in die schaduw beweegt, lijkt het donker te worden en lijkt zelfs een roodachtige kleur te zijn. Waarom rood? Omdat de atmosfeer van de Aarde het grootste deel van het blauwe licht filtert. Sommige mensen hebben het effect van de "bloedmaan" bijgenaamd. NASA zegt dat maansverduisteringen meestal ten minste twee keer per jaar plaatsvinden, maar deze verduistering is de derde in een reeks van vier op een rij, bekend als een "tetrad." De eerste was op 15 april 2014. De tweede was in september 2014, de volgende is zaterdag en er zal er een meer zijn, op 28 september. Als je meer wilt weten over de verduistering, NASA astronoom Mitzi Adam. Deel uw foto's met CNN iReport."""
@@ -36,7 +36,8 @@ class TextSummarizer:
36
 
37
  def summarize(self, model_name, input_text, generate_kwargs) -> str:
38
  if not self.generator or self.model_loaded != model_name:
39
- self.load(model_name)
 
40
  return self.generator(
41
  input_text, return_tensors=False, return_text=True, **generate_kwargs
42
  )[0].get("generated_text")
@@ -80,9 +81,10 @@ def main():
80
  """# Netherformer
81
  * Create summaries of Dutch news stories.
82
  * Copy paste any Dutch news text and press the Generate Summary botton.
83
- * This is a demo of the several Dutch T5 models finetuned for summarization:
84
  * [T5 Base Dutch finetuned on CNN DM and XSUM](https://huggingface.co/flax-community/t5-base-dutch-demo) (Jul '21)
85
  * [T5 v1.1 Base Dutch finetuned on CNN DM](https://huggingface.co/yhavinga/t5-v1.1-base-dutch-cnn-test) (Jan '22)
 
86
  """
87
  )
88
  st.sidebar.title("Parameters:")
 
11
  MODELS = [
12
  "flax-community/t5-base-dutch-demo",
13
  "yhavinga/t5-v1.1-base-dutch-cnn-test",
14
+ "yhavinga/t5-v1.1-large-dutch-cnn-test",
15
  ]
16
 
17
  DEFAULT_TEXT: str = """(CNN) Skywatchers in West-Noord-Amerika zijn in voor een traktatie: een bijna vijf minuten totale maansverduistering vanmorgen. Hier is hoe het zich ontvouwt:. Het begon om 3:16 a.m. Pacific Daylight Tijd, toen de maan begon te bewegen in de schaduw van de Aarde. Voor het volgende uur en 45 minuten, die schaduw zal bewegen over de maan en verzwolgen het om 4:58 a.m. Pacific Time. De totale verduistering zal slechts vier minuten en 43 seconden duren, en NASA zegt dat maakt het de kortste van de eeuw. Kijken live op NASA TV. Terwijl mensen ten westen van de Mississippi River zal het beste uitzicht hebben, ten minste een gedeeltelijke verduistering zal zichtbaar zijn over de hele natie. Maar zonsopgang zal de show te onderbreken op de Oostkust. Delen van Zuid-Amerika, India, China en China Een maansverduistering gebeurt wanneer de zon, de aarde en de maan een rechte lijn vormen in de ruimte, met de aarde in het midden. De zon schijnt op de Aarde en creëert een schaduw. Als de maan dieper in die schaduw beweegt, lijkt het donker te worden en lijkt zelfs een roodachtige kleur te zijn. Waarom rood? Omdat de atmosfeer van de Aarde het grootste deel van het blauwe licht filtert. Sommige mensen hebben het effect van de "bloedmaan" bijgenaamd. NASA zegt dat maansverduisteringen meestal ten minste twee keer per jaar plaatsvinden, maar deze verduistering is de derde in een reeks van vier op een rij, bekend als een "tetrad." De eerste was op 15 april 2014. De tweede was in september 2014, de volgende is zaterdag en er zal er een meer zijn, op 28 september. Als je meer wilt weten over de verduistering, NASA astronoom Mitzi Adam. Deel uw foto's met CNN iReport."""
 
36
 
37
  def summarize(self, model_name, input_text, generate_kwargs) -> str:
38
  if not self.generator or self.model_loaded != model_name:
39
+ with st.spinner(f"Loading model {model_name}..."):
40
+ self.load(model_name)
41
  return self.generator(
42
  input_text, return_tensors=False, return_text=True, **generate_kwargs
43
  )[0].get("generated_text")
 
81
  """# Netherformer
82
  * Create summaries of Dutch news stories.
83
  * Copy paste any Dutch news text and press the Generate Summary botton.
84
+ * This is a demo of several Dutch T5 models finetuned for summarization:
85
  * [T5 Base Dutch finetuned on CNN DM and XSUM](https://huggingface.co/flax-community/t5-base-dutch-demo) (Jul '21)
86
  * [T5 v1.1 Base Dutch finetuned on CNN DM](https://huggingface.co/yhavinga/t5-v1.1-base-dutch-cnn-test) (Jan '22)
87
+ * [T5 v1.1 Large Dutch finetuned on CNN DM](https://huggingface.co/yhavinga/t5-v1.1-large-dutch-cnn-test) (Jan '22)
88
  """
89
  )
90
  st.sidebar.title("Parameters:")