quarterturn
first commit
837bad6
|
raw
history blame
979 Bytes
metadata
license: cc-by-nc-4.0

Molmo 7B Flux Dev Image Captioner. Screenshot

A simple python and gradio script to use Molmo 7B for image captioning. The prompt is currently written to produce captions that work well for Flux Dev LoRA training, but you could adjust it to suit other models captioning style.

Install:

  1. create a python3 venv or use conda to create an environment, eg: conda create -n caption python=3.11

  2. activate your environment, eg: conda activate caption

  3. install the dependencies pip3 install -r requirements.txt

  4. run the gradio version: python3 main.py

    1. create a zip file of images
    2. upload it
    3. process it
    4. click the button to download the caption zip file, the link is at the top of the page

    run the command-line version: python3 caption.py

    1. make sure your images are in the "images" directory
    2. captions will be placed in the "images" directory