Upload 5 files

app.py

@@ -0,0 +1,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
+)

readme.md

@@ -0,0 +1,48 @@
+# Vietnamese Poem Generation with GPT-2
+
+This Streamlit application allows users to generate Vietnamese poems using a pre-trained GPT-2 model. The model is fine-tuned on a dataset of Vietnamese poems and is capable of generating creative and contextually relevant verses.
+
+## Features
+
+*   **Interactive Interface:** A user-friendly Streamlit interface for easy interaction.
+*   **Customizable Generation:** Users can provide a starting phrase or sentence and adjust various generation parameters like temperature, top-k, top-p, and repetition penalty.
+*   **Parameter Tuning:** Fine-tune the generation process by adjusting `temperature`, `top-k`, `top-p`, and `repetition_penalty` to achieve desired results.
+*   **Clear Output:** Displays the generated poem in a readable format, line by line.
+*   **Error Handling:** Includes error handling to gracefully manage exceptions during poem generation.
+*   **Footer Attribution:** Clear attribution to AI Vietnam and the original developer with a link to the GitHub repository.
+*   **Instructional Expander:** Provides helpful instructions on how to use the application effectively.
+
+## Usage
+
+1.  **Input Prompt:** Enter a few words or a sentence as a starting point for the poem in the text area.
+2.  **Adjust Parameters:**
+    *   **Max Output Tokens:** Set the maximum length of the generated poem.
+    *   **Temperature:** Control the randomness of the generated text. Higher values lead to more diverse and unexpected results.
+    *   **Top-k:** Limit the number of most likely next words considered at each step.
+    *   **Top-p:**  Set a probability threshold, selecting the smallest set of words whose cumulative probability exceeds the threshold.
+    *   **Repetition Penalty:**  Penalize repeated words to encourage more diverse text.
+3.  **Generate Poem:** Click the "Sinh thơ" (Generate Poem) button to generate the poem based on the provided input and parameters.
+4.  **View and Download:** The generated poem will be displayed on the screen.
+
+## Model
+
+*   The application uses the `thangduong0509/gpt2_viet_poem_generation` GPT-2 model, which has been trained specifically for generating Vietnamese poetry.
+
+## Setup
+
+1.  **Install Dependencies:**
+
+    ```bash
+    pip install streamlit transformers
+    ```
+
+2.  **Run the Application:**
+
+    ```bash
+    streamlit run app.py
+    ```
+
+## Credits
+
+*   Developed by [Koii2k3(https://github.com/Koii2k3/Poem-Generation) for AI Vietnam.
+*   GPT-2 Model: `thangduong0509/gpt2_viet_poem_generation`

requirements.txt

@@ -0,0 +1,3 @@
+streamlit==1.41.1
+transformers==4.47.0
+torch==2.5.1