Update README.md

README.md

@@ -15,7 +15,38 @@ should probably proofread and complete it, then remove this comment. -->
 
 # paligemma_age
 
-This model is a fine-tuned version of [google/paligemma-3b-pt-224](https://huggingface.co/google/paligemma-3b-pt-224) on the imagefolder dataset.
+This model is a fine-tuned version of [google/paligemma-3b-pt-224](https://huggingface.co/google/paligemma-3b-pt-224) on the FairFace dataset.
+
+
+``` python
+
+from PIL import Image
+import torch
+from transformers import PaliGemmaProcessor, PaliGemmaForConditionalGeneration, BitsAndBytesConfig, TrainingArguments, Trainer
+
+
+model = PaliGemmaForConditionalGeneration.from_pretrained('NYUAD-ComNets/paligemma_age',torch_dtype=torch.bfloat16)
+
+input_text = "what is age group of person in the image?"
+
+processor = PaliGemmaProcessor.from_pretrained("google/paligemma-3b-pt-224")
+
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
+model.to(device)
+
+
+input_image = Image.open('image_path')
+inputs = processor(text=input_text, images=input_image, padding="longest", do_convert_rgb=True, return_tensors="pt").to(device)
+inputs = inputs.to(dtype=model.dtype)
+      
+with torch.no_grad():
+          output = model.generate(**inputs, max_length=500)
+result=processor.decode(output[0], skip_special_tokens=True)[len(input_text):].strip()
+
+
+```
+
 
 ## Model description