pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2026-02-06 18:45:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor: Improve Picture.py resource usage (#13011)

Browse Source 
### What problem does this PR solve?

Improve Picture.py resource usage

### Type of change


- [x] Refactoring
This commit is contained in:
Stephen Hu
2026-02-06 09:50:53 +08:00
committed by GitHub
parent 1262533b74
commit 11703d957d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
rag/app/picture.py
View File

@ -78,10 +78,10 @@ def chunk(filename, binary, tenant_id, lang, callback=None, **kwargs):
try: try:
callback(0.4, "Use CV LLM to describe the picture.") callback(0.4, "Use CV LLM to describe the picture.")
cv_mdl = LLMBundle(tenant_id, LLMType.IMAGE2TEXT, lang=lang) cv_mdl = LLMBundle(tenant_id, LLMType.IMAGE2TEXT, lang=lang)
img_binary = io.BytesIO() with io.BytesIO() as img_binary:
img.save(img_binary, format="JPEG") img.save(img_binary, format="JPEG")
img_binary.seek(0) img_binary.seek(0)
ans = cv_mdl.describe(img_binary.read()) ans = cv_mdl.describe(img_binary.read())
callback(0.8, "CV LLM respond: %s ..." % ans[:32]) callback(0.8, "CV LLM respond: %s ..." % ans[:32])
txt += "\n" + ans txt += "\n" + ans
tokenize(doc, txt, eng) tokenize(doc, txt, eng)