Update modeling_GOT.py

modeling_GOT.py

@@ -580,7 +580,8 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
             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
@@ -655,7 +656,7 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
 
                     with smart_open(html_path_2, 'w') as web_f_new:
                         web_f_new.write(new_web)
-        return output_ids
+        return response_str
 
     def dynamic_preprocess(self, image, min_num=1, max_num=6, image_size=1024, use_thumbnail=True):
         
@@ -818,6 +819,7 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
         if outputs.endswith(stop_str):
             outputs = outputs[:-len(stop_str)]
         outputs = outputs.strip()   
+        response_str = outputs
 
         if render:
             print('==============rendering===============')
@@ -846,4 +848,4 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
             with smart_open(html_path_2, 'w') as web_f_new:
                 web_f_new.write(new_web)
 
-        return output_ids
+        return response_str