python - 封装moondream（备份）

目录

一、 moondream封装成类

二、moondream封装成http api

---

一、 moondream封装成类

# moondream_model.py
#https://github.com/vikhyat/moondream?tab=readme-ov-fileimport moondream as md
from PIL import Imageclass 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_imagedef generate_caption(self, encoded_image):# 生成