upload sample png and update readme

.gitattributes

@@ -34,3 +34,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 tokenizer.json filter=lfs diff=lfs merge=lfs -text
+*.json filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -56,13 +56,10 @@ from InfiMed import InfiMed
 from PIL import Image
 import torch
 
-# Define the path to the pretrained checkpoint
-pretrained_model_path = "."
-
 # Load the model from the pretrained checkpoint
-model = InfiMed.from_pretrained(pretrained_model_path, device_map="auto", torch_dtype=torch.bfloat16)
+model = InfiMed.from_pretrained("InfiX-ai/InfiMed-Foundation-4B", device_map="auto", torch_dtype=torch.bfloat16)
 
-image_path = ""  # Replace with the path to your image file
+image_path = "sample.png"  # Replace with the path to your image file
 image = Image.open(image_path).convert("RGB")  # Ensure the image is in RGB format
 
 # Prepare input messages