moondream-docker-api/start_moondream.py

21 lines
386 B
Python

import os
from dotenv import load_dotenv
import moondream as md
from PIL import Image
# Load .env
load_dotenv()
# Get key from env
api_key = os.getenv("MOON_DREAM_KEY")
# Use Moondream cloud model
model = md.vl(api_key=api_key)
# Load image
image = Image.open("nat.jpg")
# Generate caption
caption_response = model.caption(image, length="short")
print(caption_response["caption"])