Update README.md

README.md

@@ -2,7 +2,7 @@
 license: mit
 language:
 - en
-pipeline_tag: question-answering
+pipeline_tag: image-to-text
 ---
 # git_20
 
@@ -13,7 +13,7 @@ This model is fine-tuned with LLaMA with 8 Nvidia A100-80G GPUs using 3,000,000
 ```python
 from transformers import AutoModelForCausalLM
 from transformers import AutoProcessor
-
+from PIL import Image
 model = AutoModelForCausalLM.from_pretrained("Fan21/git_20")
 processor = AutoProcessor.from_pretrained("Fan21/git_20")
 
@@ -26,4 +26,4 @@ with torch.no_grad():
     outputs = model.generate(pixel_values=pixel_values, max_length=50)
 
 answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
-```
+```