111

README.md

@@ -4,7 +4,7 @@ license: apache-2.0
 # Diagram Formalization Enhanced Multi-Modal Geometry Problem Solver
 ## Model Structure
 
-<img src="sample/DFE-GPS.png" alt="Alt text" width="30%" height="auto">
+<img src="sample/DFE-GPS.png" alt="Alt text" width="80%" height="auto">
 
 
 - **Diagram Encoder**: [siglip-so400m-patch14-384](https://huggingface.co/google/siglip-so400m-patch14-384)
@@ -18,7 +18,7 @@ Before running the script, install the following necessary dependencies.
 pip install torch transformers==4.40.0 accelerate pillow sentencepiece
 ```
 
-You can solve geometric problems using the following script. First, formalize the geometric images with the Diagram Formalizer, and then use the multi-modal reasing model for problem-solving:
+You can solve geometric problems using the following script. First, formalize the geometric images with the [Diagram Formalizer](https://huggingface.co/NaughtyDog97/DiagramFormalizer), and then use the multi-modal reasing model for problem-solving:
 
 ```python
 import torch
@@ -65,7 +65,6 @@ def parse_cdl(input_string):
         if match:
             results[key] = match.group(1).strip()
         else:
-            # 如果未找到包含"calibrate"的版本，尝试匹配不含"calibrate"的版本
             pattern = pattern.replace("(?:calibrate )", "(?:calibrate )?")
             match = re.search(pattern, input_string, re.DOTALL)
             if match: