Update README.md

README.md

@@ -32,6 +32,7 @@ model.get_gradio_demo(tokenizer, image_processor).queue(max_size=20).launch(serv
 ```python
 import os; os.system("pip3 install transformers accelerate bitsandbytes gradio fire")
 from transformers import AutoModel, AutoTokenizer, AutoImageProcessor
+import torch
 
 model_id = "jxu124/TiO"
 model = AutoModel.from_pretrained(model_id, trust_remote_code=True, torch_dtype=torch.float16).cuda()
@@ -41,7 +42,6 @@ image_processor = AutoImageProcessor.from_pretrained(model_id)
 # ---- mini example ----
 from PIL import Image
 from io import BytesIO
-import torch
 import requests
 
 # Prepare example