Update README.md

README.md

@@ -8,7 +8,6 @@ language:
 pipeline_tag: image-to-text
 ---
 
-
 ## Model Details
 
 ### Model Description
@@ -16,8 +15,6 @@ pipeline_tag: image-to-text
 - **Developed by:** [https://huggingface.co/Mit1208]
 - **Finetuned from model:** [microsoft/kosmos-2-patch14-224]
 
-[More Information Needed]
-
 ## Training Details
 https://github.com/mit1280/fined-tuning/blob/main/Kosmos_2_fine_tune_PokemonCards_trl.ipynb
 
@@ -26,10 +23,13 @@ https://github.com/mit1280/fined-tuning/blob/main/kosmos2_fine_tuned_inference.i
 
 ### How to Use
 ```python
-# Load model directly
 from transformers import AutoProcessor, Kosmos2ForConditionalGeneration
+import torch
+from io import BytesIO
+import requests
+from PIL import Image
 
-# processor = AutoProcessor.from_pretrained("Mit1208/Kosmos-2-PokemonCards-trl-merged")
+processor = AutoProcessor.from_pretrained("microsoft/kosmos-2-patch14-224")
 my_model = Kosmos2ForConditionalGeneration.from_pretrained("Mit1208/Kosmos-2-PokemonCards-trl-merged", device_map="auto",low_cpu_mem_usage=True)
 
 # load image