OLIVE2403 commited on
Commit
5120511
·
verified ·
1 Parent(s): d90a0e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ st.set_page_config(page_title="Brand Identity Generator", layout="wide")
8
  # Load FLAN-T5 model pipeline
9
  @st.cache_resource
10
  def load_model():
11
- return pipeline("text2text-generation", model="google/flan-t5-large")
12
 
13
  generator = load_model()
14
 
@@ -40,7 +40,7 @@ st.markdown(
40
  st.markdown(
41
  """
42
  Generate **brand names**, **taglines**, **personality descriptions**, **colour palettes**, **target audiences**, and **slogans**
43
- for your startup idea using the powerful **FLAN-T5-LARGE** model.
44
  """
45
  )
46
 
 
8
  # Load FLAN-T5 model pipeline
9
  @st.cache_resource
10
  def load_model():
11
+ return pipeline("text2text-generation", model="google/flan-t5-base", device=-1)
12
 
13
  generator = load_model()
14
 
 
40
  st.markdown(
41
  """
42
  Generate **brand names**, **taglines**, **personality descriptions**, **colour palettes**, **target audiences**, and **slogans**
43
+ for your startup idea using the powerful **FLAN-T5-BASE** model.
44
  """
45
  )
46