zRzRzRzRzRzRzR commited on
Commit
4fa9584
·
1 Parent(s): 79d57f6
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -4,7 +4,6 @@ from transformers import AutoProcessor, Glm4vForConditionalGeneration, TextItera
4
  from pathlib import Path
5
  import threading
6
  import re
7
- import argparse
8
  import copy
9
  import spaces
10
  import fitz
@@ -80,7 +79,7 @@ class GLM4VModel:
80
  seg = re.search(r"<think>(.*?)</think>", buf, re.DOTALL)
81
  if seg:
82
  think_html = (
83
- "<details open><summary style='cursor:pointer;font-weight:bold;color:#bbbbbb;'>💭 思考过程</summary>"
84
  "<div style='color:#cccccc;line-height:1.4;padding:10px;border-left:3px solid #666;margin:5px 0;background-color:rgba(128,128,128,0.1);'>"
85
  + seg.group(1).strip().replace("\n", "<br>")
86
  + "</div></details>"
@@ -88,7 +87,7 @@ class GLM4VModel:
88
  else:
89
  part = buf.split("<think>", 1)[1]
90
  think_html = (
91
- "<details open><summary style='cursor:pointer;font-weight:bold;color:#bbbbbb;'>💭 思考过程</summary>"
92
  "<div style='color:#cccccc;line-height:1.4;padding:10px;border-left:3px solid #666;margin:5px 0;background-color:rgba(128,128,128,0.1);'>"
93
  + part.replace("\n", "<br>")
94
  + "</div></details>"
 
4
  from pathlib import Path
5
  import threading
6
  import re
 
7
  import copy
8
  import spaces
9
  import fitz
 
79
  seg = re.search(r"<think>(.*?)</think>", buf, re.DOTALL)
80
  if seg:
81
  think_html = (
82
+ "<details open><summary style='cursor:pointer;font-weight:bold;color:#bbbbbb;'>💭 Thinking</summary>"
83
  "<div style='color:#cccccc;line-height:1.4;padding:10px;border-left:3px solid #666;margin:5px 0;background-color:rgba(128,128,128,0.1);'>"
84
  + seg.group(1).strip().replace("\n", "<br>")
85
  + "</div></details>"
 
87
  else:
88
  part = buf.split("<think>", 1)[1]
89
  think_html = (
90
+ "<details open><summary style='cursor:pointer;font-weight:bold;color:#bbbbbb;'>💭 Thinking</summary>"
91
  "<div style='color:#cccccc;line-height:1.4;padding:10px;border-left:3px solid #666;margin:5px 0;background-color:rgba(128,128,128,0.1);'>"
92
  + part.replace("\n", "<br>")
93
  + "</div></details>"