Update README.md

README.md

@@ -17,12 +17,8 @@ tags:
 For model inference, please refer to the following example code:
 
 ```python
-from transformers import (AutoTokenizer, 
-                          AutoModelForCausalLM, 
-                          BitsAndBytesConfig, 
-                          Trainer, 
-                          TrainingArguments,
-                          pipeline)
+import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
 model_path = 'yueqingyou/BioQwen-1.8B'
 tokenizer = AutoTokenizer.from_pretrained(model_path, use_fast=True)