Update README.md

README.md

@@ -12,32 +12,36 @@ language: zh
 ## 代码调用
 
 ### forward
-> **Parameters**
-> * `input_ids: Optional[torch.Tensor] = None`
-> * `token_type_ids: Optional[torch.Tensor] = None`
-> * `position_ids: Optional[torch.Tensor] = None`
-> * `attention_mask: Optional[torch.Tensor] = None`
-> * `head_mask: Optional[torch.Tensor] = None`
-> * `inputs_embeds: Optional[torch.Tensor] = None`
-> * `start_positions: Optional[torch.Tensor] = None`
-> * `end_positions: Optional[torch.Tensor] = None`
-> * `output_attentions: Optional[bool] = None`
-> * `output_hidden_states: Optional[bool] = None`
-> * `return_dict: Optional[bool] = None`
-> 
-> **Returns**  *UIEModelOutput or tuple(torch.FloatTensor)*
+#### *Parameters*
+* `input_ids: Optional[torch.Tensor] = None`
+* `token_type_ids: Optional[torch.Tensor] = None`
+* `position_ids: Optional[torch.Tensor] = None`
+* `attention_mask: Optional[torch.Tensor] = None`
+* `head_mask: Optional[torch.Tensor] = None`
+* `inputs_embeds: Optional[torch.Tensor] = None`
+* `start_positions: Optional[torch.Tensor] = None`
+* `end_positions: Optional[torch.Tensor] = None`
+* `output_attentions: Optional[bool] = None`
+* `output_hidden_states: Optional[bool] = None`
+* `return_dict: Optional[bool] = None`
+
+#### *Returns*  
+
+* `UIEModelOutput or tuple(torch.FloatTensor)`
 
 ### predict
-> **Parameters**
-> * `schema: Union[Dict, List[str], str]`
-> * `input_texts: Union[List[str], str]`
-> * `tokenizer: PreTrainedTokenizerFast`
-> * `max_length: int = 512`
-> * `batch_size: int = 32`
-> * `position_prob: int = 0.5`
-> * `progress_hook=None`
->
-> **Returns**  * List[Dict]*
+#### *Parameters*
+* `schema: Union[Dict, List[str], str]`
+* `input_texts: Union[List[str], str]`
+* `tokenizer: PreTrainedTokenizerFast`
+* `max_length: int = 512`
+* `batch_size: int = 32`
+* `position_prob: int = 0.5`
+* `progress_hook=None`
+
+#### *Returns*  
+* `List[Dict]`
+
 
 ```python
 from tqdm import tqdm