Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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-
|
| 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-
|
| 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 |
|