Change eval to ast.literal_eval

modeling_deepseekocr.py

@@ -2,6 +2,7 @@
 import os
 import math
 import re
+import ast
 from tqdm import tqdm
 from abc import ABC
 from typing import List, Optional, Tuple, Union
@@ -66,7 +67,7 @@ def extract_coordinates_and_label(ref_text, image_width, image_height):
 
     try:
         label_type = ref_text[1]
-        cor_list = eval(ref_text[2])
+        cor_list = ast.literal_eval(ref_text[2])
     except Exception as e:
         print(e)
         return None
@@ -1003,12 +1004,12 @@ class DeepseekOCRForCausalLM(DeepseekV2ForCausalLM):
 
             if 'line_type' in outputs:
                 import matplotlib.pyplot as plt
-                lines = eval(outputs)['Line']['line']
+                lines = ast.literal_eval(outputs)['Line']['line']
 
-                line_type = eval(outputs)['Line']['line_type']
+                line_type = ast.literal_eval(outputs)['Line']['line_type']
                 # print(lines)
 
-                endpoints = eval(outputs)['Line']['line_endpoint']
+                endpoints = ast.literal_eval(outputs)['Line']['line_endpoint']
 
                 fig, ax = plt.subplots(figsize=(3,3), dpi=200)
                 ax.set_xlim(-15, 15)
@@ -1016,8 +1017,8 @@ class DeepseekOCRForCausalLM(DeepseekV2ForCausalLM):
 
                 for idx, line in enumerate(lines):
                     try:
-                        p0 = eval(line.split(' -- ')[0])
-                        p1 = eval(line.split(' -- ')[-1])
+                        p0 = ast.literal_eval(line.split(' -- ')[0])
+                        p1 = ast.literal_eval(line.split(' -- ')[-1])
 
                         if line_type[idx] == '--':
                             ax.plot([p0[0], p1[0]], [p0[1], p1[1]], linewidth=0.8, color='k')
@@ -1032,7 +1033,7 @@ class DeepseekOCRForCausalLM(DeepseekV2ForCausalLM):
                 for endpoint in endpoints:
 
                     label = endpoint.split(': ')[0]
-                    (x, y) = eval(endpoint.split(': ')[1])
+                    (x, y) = ast.literal_eval(endpoint.split(': ')[1])
                     ax.annotate(label, (x, y), xytext=(1, 1), textcoords='offset points', 
                                 fontsize=5, fontweight='light')