imagestory.ai

imagestory.ai is a simple AI website that writes stories based on a given image.

Uses Flask for back-end, Bootstrap for front-end.

Uses nlpconnect/vit-gpt2-image-captioning for image-to-text, TheBloke/orca_mini_3B-GGML for text generation.

Installation & usage

• (Optional) Create a virtual environment

Linux:

python -m venv .venv
source .venv/bin/activate

Windows:

python -m venv .venv
.\.venv\Scripts\activate

If there is error with "Execution Policies", run:

Set-ExecutionPolicy -Scope CurrentUser -ExecutionPolicy Unrestricted

• Use the package manager pip to install require packages.

pip install -r requirements.txt

• Create folder static/uploads

mkdir static/uploads

• To run:

flask run 

Project Structure

• templates/: html files
• static/: images, uploaded files

Contributing

Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.