Update README.md

README.md

@@ -47,8 +47,8 @@ model = AutoModelForCausalLM.from_pretrained("thwri/CogFlorence-2-Large-Freeze",
 processor = AutoProcessor.from_pretrained("thwri/CogFlorence-2-Large-Freeze", trust_remote_code=True)
 
 # Function to run the model on an example
-def run_example(task_prompt, text_input, image):
-    prompt = task_prompt + text_input
+def run_example(task_prompt, image):
+    prompt = task_prompt
 
     # Ensure the image is in RGB mode
     if image.mode != "RGB":
@@ -72,7 +72,8 @@ import copy
 
 url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/car.jpg?download=true"
 image = Image.open(requests.get(url, stream=True).raw)
-run_example("<MORE_DETAILED_CAPTION>", "Describe this image in great detail.", image)
+result = run_example("<MORE_DETAILED_CAPTION>" , image)
+print(result)
 
 # {'<MORE_DETAILED_CAPTION>': 'a turquoise volkswagen beetle parked on a cobblestone street in front of a yellow wall with two wooden doors. the car's body is painted in a vibrant shade of teal, with a glossy finish that reflects the sunlight, and the wheels are polished with a silver hubcap. the building behind the car has a weathered, aged appearance, with visible cracks and peeling paint. the sky above is clear and blue, suggesting a sunny day.'}
 ```