LSTM Text Generator

A web application for text generation using LSTM neural networks, built with Svelte 5 and TensorFlow.js.

Features

• 🧠 Train LSTM model on custom text input
• ✨ Generate text based on trained model
• 🚀 GPU acceleration support (WebGL) for faster training
• 💻 Simple and responsive user interface
• 📊 Real-time training progress display

Technologies

• Svelte 5 - Reactive framework
• TypeScript - Type safety
• TensorFlow.js - Machine learning in the browser
• Vite - Fast build tool and dev server

Installation

# Install dependencies
npm install

# Run development server
npm run dev

# Build for production
npm run build

# Preview production build
npm run preview

Usage

1. Enter your training text in the "Training Data" section (minimum 10 characters)
2. Click "Train Model" and wait for training to complete
3. Enter a seed text in the "Prediction" section
4. Click "Generate Text" to generate new text

Project Structure

src/
├── lib/
│   ├── modelService.ts      # Training and generation logic
│   ├── modelUtils.ts        # Utility functions
│   ├── tfConfig.ts          # TensorFlow configuration
│   ├── TrainingSection.svelte
│   ├── PredictionSection.svelte
│   └── OutputDisplay.svelte
├── App.svelte               # Main component
└── main.ts                  # Entry point

Notes

• For better results, use longer training texts (multiple paragraphs)
• The model uses two LSTM layers with dropout to prevent overfitting
• GPU acceleration is automatically used if available