File size: 3,027 Bytes
f550617 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 |
import streamlit as st
from transformers import pipeline
st.set_page_config(
page_title="AIVN - Vietnamese Poem Generation",
page_icon="./static/aivn_favicon.png",
layout="wide",
initial_sidebar_state="expanded"
)
st.image("./static/aivn_logo.png", width=300)
st.title("Vietnamese Poem Generation")
st.write("The used model is GPT-2 trained on Vietnamese poems: thangduong0509/gpt2_viet_poem_generation")
with st.expander("Instructions"):
st.write("""
1. Enter a few words or sentences as a suggestion for the poem.
2. Adjust the parameters to change the way the poem is generated:
- Temperature: Higher values produce more diverse results.
- Top-k: The number of words with the highest probability to choose from at each step.
- Top-p: Cumulative probability threshold for word selection.
- Repetition penalty: Adjust to avoid repetition.
3. Press the "Generate Poem" button to create the poem.
4. You can download the generated poem using the "Download Poem" button.
""")
prompt_input = st.text_area("Enter a few words or sentences to start:",
value="Con song que toi dep\n",
height=100)
col1, empty_col, col2 = st.columns([0.7, 0.3, 1.0])
with col1:
max_length = st.slider("Max Output Tokens:", 10, 200, 75)
temperature = st.slider("Temperature:", 0.1, 1.5, 0.8)
top_k = st.slider("Top-k:", 1, 100, 50)
top_p = st.slider("Top-p:", 0.1, 1.0, 0.95)
repetition_penalty = st.slider("Repetition Penalty:", 1.0, 2.0, 1.2)
with empty_col:
st.empty()
with col2:
if st.button("Generate Poem"):
with st.spinner("Generating poem..."):
try:
# Initialize model
generator = pipeline('text-generation',
model='thangduong0509/gpt2_viet_poem_generation')
# Generate poem
results = generator(
prompt_input,
max_new_tokens=max_length,
do_sample=True,
top_k=top_k,
top_p=top_p,
temperature=temperature,
repetition_penalty=repetition_penalty
)
generated_text = results[0]['generated_text']
st.subheader("Generated Poem:")
for line in generated_text.split('\n'):
st.write(line)
except Exception as e:
st.error(f"An error occurred: {str(e)}")
st.markdown(
"""
<style>
.footer {
position: fixed;
bottom: 0;
left: 0;
width: 100%;
background-color: #f1f1f1;
text-align: center;
padding: 10px 0;
font-size: 14px;
color: #555;
}
</style>
<div class="footer">
2024 AI VIETNAM | Made by <a href="https://github.com/Koii2k3/Poem-Generation" target="_blank">Koii2k3</a>
</div>
""",
unsafe_allow_html=True
) |