Update modeling_GOT.py

modeling_GOT.py

@@ -590,84 +590,84 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
                     stopping_criteria=[stopping_criteria]
                     )
             
-            outputs = tokenizer.decode(output_ids[0, input_ids.shape[1]:]).strip()
-            
-            if outputs.endswith(stop_str):
-                outputs = outputs[:-len(stop_str)]
-            outputs = outputs.strip()
-            response_str = outputs
-
-            if render:
-                print('==============rendering===============')
-                from .render_tools import svg_to_html, content_mmd_to_html, tik_html, translation_table
-
-                if '**kern' in outputs:
-                    import verovio
-                    tk = verovio.toolkit()
-                    tk.loadData(outputs)
-                    tk.setOptions({"pageWidth": 2100, "footer": 'none',
-                'barLineWidth': 0.5, 'beamMaxSlope': 15,
-                'staffLineWidth': 0.2, 'spacingStaff': 6})
-                    tk.getPageCount()
-                    svg = tk.renderToSVG()
-                    svg = svg.replace("overflow=\"inherit\"", "overflow=\"visible\"")
-
-                    svg_to_html(svg, save_render_file)
-
-                if ocr_type == 'format' and '**kern' not in outputs:
+        outputs = tokenizer.decode(output_ids[0, input_ids.shape[1]:]).strip()
+        
+        if outputs.endswith(stop_str):
+            outputs = outputs[:-len(stop_str)]
+        outputs = outputs.strip()
+        response_str = outputs
 
-                    
-                    if  '\\begin{tikzpicture}' not in outputs:
-                        html_path_2 = save_render_file
-                        right_num = outputs.count('\\right')
-                        left_num = outputs.count('\left')
+        if render:
+            print('==============rendering===============')
+            from .render_tools import svg_to_html, content_mmd_to_html, tik_html, translation_table
 
-                        if right_num != left_num:
-                            outputs = outputs.replace('\left(', '(').replace('\\right)', ')').replace('\left[', '[').replace('\\right]', ']').replace('\left{', '{').replace('\\right}', '}').replace('\left|', '|').replace('\\right|', '|').replace('\left.', '.').replace('\\right.', '.')
+            if '**kern' in outputs:
+                import verovio
+                tk = verovio.toolkit()
+                tk.loadData(outputs)
+                tk.setOptions({"pageWidth": 2100, "footer": 'none',
+            'barLineWidth': 0.5, 'beamMaxSlope': 15,
+            'staffLineWidth': 0.2, 'spacingStaff': 6})
+                tk.getPageCount()
+                svg = tk.renderToSVG()
+                svg = svg.replace("overflow=\"inherit\"", "overflow=\"visible\"")
 
+                svg_to_html(svg, save_render_file)
 
-                        outputs = outputs.replace('"', '``').replace('$', '')
+            if ocr_type == 'format' and '**kern' not in outputs:
 
-                        outputs_list = outputs.split('\n')
-                        gt= ''
-                        for out in outputs_list:
-                            gt +=  '"' + out.replace('\\', '\\\\') + r'\n' + '"' + '+' + '\n' 
-                        
-                        gt = gt[:-2]
-
-
-                        lines = content_mmd_to_html
-                        lines = lines.split("const text =")
-                        new_web = lines[0] + 'const text ='  + gt  + lines[1]
-
-                    else:
-                        html_path_2 = save_render_file
-                        outputs = outputs.translate(translation_table)
-                        outputs_list = outputs.split('\n')
-                        gt= ''
-                        for out in outputs_list:
-                            if out:
-                                if '\\begin{tikzpicture}' not in out and '\\end{tikzpicture}' not in out:
-                                    while out[-1] == ' ':
-                                        out = out[:-1]
-                                        if out is None:
-                                            break
-        
-                                    if out:
-                                        if out[-1] != ';':
-                                            gt += out[:-1] + ';\n'
-                                        else:
-                                            gt += out + '\n'
-                                else:
-                                    gt += out + '\n'
+                
+                if  '\\begin{tikzpicture}' not in outputs:
+                    html_path_2 = save_render_file
+                    right_num = outputs.count('\\right')
+                    left_num = outputs.count('\left')
+
+                    if right_num != left_num:
+                        outputs = outputs.replace('\left(', '(').replace('\\right)', ')').replace('\left[', '[').replace('\\right]', ']').replace('\left{', '{').replace('\\right}', '}').replace('\left|', '|').replace('\\right|', '|').replace('\left.', '.').replace('\\right.', '.')
+
+
+                    outputs = outputs.replace('"', '``').replace('$', '')
+
+                    outputs_list = outputs.split('\n')
+                    gt= ''
+                    for out in outputs_list:
+                        gt +=  '"' + out.replace('\\', '\\\\') + r'\n' + '"' + '+' + '\n' 
+                    
+                    gt = gt[:-2]
+
+
+                    lines = content_mmd_to_html
+                    lines = lines.split("const text =")
+                    new_web = lines[0] + 'const text ='  + gt  + lines[1]
+
+                else:
+                    html_path_2 = save_render_file
+                    outputs = outputs.translate(translation_table)
+                    outputs_list = outputs.split('\n')
+                    gt= ''
+                    for out in outputs_list:
+                        if out:
+                            if '\\begin{tikzpicture}' not in out and '\\end{tikzpicture}' not in out:
+                                while out[-1] == ' ':
+                                    out = out[:-1]
+                                    if out is None:
+                                        break
+    
+                                if out:
+                                    if out[-1] != ';':
+                                        gt += out[:-1] + ';\n'
+                                    else:
+                                        gt += out + '\n'
+                            else:
+                                gt += out + '\n'
 
 
-                        lines = tik_html
-                        lines = lines.split("const text =")
-                        new_web = lines[0] + gt + lines[1]
+                    lines = tik_html
+                    lines = lines.split("const text =")
+                    new_web = lines[0] + gt + lines[1]
 
-                    with smart_open(html_path_2, 'w') as web_f_new:
-                        web_f_new.write(new_web)
+                with smart_open(html_path_2, 'w') as web_f_new:
+                    web_f_new.write(new_web)
         return response_str
 
     def dynamic_preprocess(self, image, min_num=1, max_num=6, image_size=1024, use_thumbnail=True):