Update README.md

README.md

@@ -9,7 +9,7 @@ model-index:
 language:
 - en
 library_name: transformers
-pipeline_tag: image-to-text
+pipeline_tag: text-generation
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -17,21 +17,44 @@ should probably proofread and complete it, then remove this comment. -->
 
 # git-base-on-diffuision-dataset2
 
-This model is a fine-tuned version of [microsoft/git-base](https://huggingface.co/microsoft/git-base) on an unknown dataset.
+This model is a fine-tuned version of [microsoft/git-base](https://huggingface.co/microsoft/git-base) on hieudinhpro/diffuision-dataset2 dataset.
 
 ## Model description
 
-More information needed
 
-## Intended uses & limitations
 
-More information needed
+## How to use mdoel
 
-## Training and evaluation data
 
-More information needed
+```
+# Load model directly
+from transformers import AutoProcessor, AutoModelForCausalLM
+
+processor = AutoProcessor.from_pretrained("microsoft/git-base")
+model = AutoModelForCausalLM.from_pretrained("hieudinhpro/git-base-on-diffuision-dataset2")
+
+```
+
+```
+# load image
+from PIL import Image
+
+image = Image.open('/content/image_3.jpg')
+```
+```
+# pre image
+inputs = processor(images=image, return_tensors="pt")
+pixel_values = inputs.pixel_values
+
+# predict 
+generated_ids = model.generate(pixel_values=pixel_values, max_length=50)
+
+# decode to text
+generated_caption = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
+print(generated_caption)
+```
+
 
-## Training procedure
 
 ### Training hyperparameters
 
@@ -46,8 +69,6 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 1
 
-### Training results
-
 
 
 ### Framework versions