python - 封装moondream（备份）

目录

一、 moondream封装成类

二、moondream封装成http api

---

一、 moondream封装成类

# moondream_model.py
#https://github.com/vikhyat/moondream?tab=readme-ov-file


import moondream as md
from PIL import Image

class MoondreamModel:
    def __init__(self, model_path):
        # 初始化 Moondream 模型，加载本地模型文件
        self.model = md.vl(model=model_path)

    def encode_image(self, image_path):
        # 加载并处理图片
        image = Image.open(image_path)
        # 对图片进行编码
        encoded_image = self.model.encode_image(image)
        return encoded_image

    def generate_caption(self, encoded_image):
        # 生成