README.md

metadata

title: SmolLM2 Text Generator
emoji: 🦀
colorFrom: blue
colorTo: gray
sdk: gradio
sdk_version: 5.12.0
app_file: app.py
pinned: false

SmolLM2 Text Generator

This is a Gradio application for generating text using the trained SmolLM2 model. The app allows users to input a text prompt and generate multiple sequences of text based on that prompt. The number of sequences and the length of the generated text can be adjusted using sliders.

Features

• Text Generation: Generate text based on a user-provided prompt using the SmolLM2 model.
• Adjustable Length: Control the length of the generated text.
• Multiple Sequences: Generate multiple sequences of text in one go.

Requirements

To run this application, you need the following Python packages:

• torch
• transformers
• gradio

You can install the required packages using pip:

pip install -r requirements.txt

Usage

1. Run the App: Launch the Gradio app by running the following command in your terminal:

   python app.py
   

2. Input Prompt: Enter your desired text prompt in the provided textbox.

3. Adjust Sliders:

   • Use the "Predict Additional Text of Length" slider to set the desired length of the generated text.
   • Use the "Number of Sequences to Generate" slider to specify how many sequences you want to generate.

4. Generate Text: Click the "Generate Text" button to produce the text sequences.

5. View Output: The generated sequences will be displayed in the output textbox, each prefixed with "Sequence X:" for clarity.

Example

• Prompt: "Once upon a time"
• Number of Sequences: 2

Output:

Sequence 1:
Once upon a time, there is a cat ....

Sequence 2:
Once upon a time in a small village ....

License

This project is licensed under the MIT License. See the LICENSE file for more details.

Acknowledgments

• Hugging Face for the Transformers library and model support.
• Gradio for providing an easy-to-use interface for machine learning applications.
• The SmolLM2 model for enabling advanced text generation capabilities.