浏览代码

Refactor:Introduce Image Close For GeminiCV (#9147)

### What problem does this PR solve?

Introduce Image Close For GeminiCV

### Type of change

- [x] Refactoring
- [x] Performance Improvement
tags/v0.20.0
Stephen Hu 3 个月前
父节点
当前提交
5ccdb95008
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 2 次插入0 次删除
  1. 2
    0
      rag/llm/cv_model.py

+ 2
- 0
rag/llm/cv_model.py 查看文件

@@ -488,6 +488,7 @@ class GeminiCV(Base):
img = open(BytesIO(base64.b64decode(b64)))
input = [prompt, img]
res = self.model.generate_content(input)
img.close()
return res.text, res.usage_metadata.total_token_count

def describe_with_prompt(self, image, prompt=None):
@@ -500,6 +501,7 @@ class GeminiCV(Base):
res = self.model.generate_content(
input,
)
img.close()
return res.text, res.usage_metadata.total_token_count

def chat(self, system, history, gen_conf, images=[]):

正在加载...
取消
保存