This project is a simple web application that allows users to load pre-trained models from Hugging Face's model hub and generate text based on a user-provided prompt. It's built using Streamlit and the transformers library.
- Load Hugging Face models dynamically by specifying the model name.
- Generate text using the loaded model with a custom prompt.
- Keep a history of loaded models for quick access.
Make sure you have Python 3.6 or later installed. You can then install the dependencies with pip:
-
python -m venv .venv -
Windows:
.venv\Scripts\activate
Linux:source .venv/bin/activate -
pip install -r requirements
To run the app, navigate to the directory containing index.py and run:
streamlit run index.py
or
python start.py
The Streamlit web interface should open in your default web browser, where you can interact with the model.
Contributions are welcome! If you have a suggestion or an issue, please open an issue on the repository to discuss it.
MIT © simwai