deepgeek/MiniCPM-o
2
0
Fork 0
mirror of https://github.com/OpenBMB/MiniCPM-V.git synced 2026-02-05 18:29:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix multiple rounds of conversations in web_demo

Browse Source
This commit is contained in:
Hongji Zhu
2024-05-29 11:37:23 +08:00
parent 573844b146
commit e88df8f04d
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
web_demo.py
View File

@@ -154,7 +154,7 @@ def chat(img, msgs, ctx, params=None, vision_hidden_states=None):
res = res.replace('</ref>', '') res = res.replace('</ref>', '')
res = res.replace('<box>', '') res = res.replace('<box>', '')
answer = res.replace('</box>', '') answer = res.replace('</box>', '')
return -1, answer, None, None return 0, answer, None, None
except Exception as err: except Exception as err:
print(err) print(err)
traceback.print_exc() traceback.print_exc()

2
web_demo_2.5.py
View File

@@ -151,7 +151,7 @@ def chat(img, msgs, ctx, params=None, vision_hidden_states=None):
res = res.replace('</ref>', '') res = res.replace('</ref>', '')
res = res.replace('<box>', '') res = res.replace('<box>', '')
answer = res.replace('</box>', '') answer = res.replace('</box>', '')
return -1, answer, None, None return 0, answer, None, None
except Exception as err: except Exception as err:
print(err) print(err)
traceback.print_exc() traceback.print_exc()