Update README.md

README.md

@@ -11,43 +11,42 @@ tags:
 - text-generation-inference
 - label
 ---
-![10.png](https://cdn-uploads.huggingface.co/production/uploads/65bb837dbfb878f46c77de4c/tCeb4MVz79GCiOmMxkLcN.png)
-# **LabelPRO**
+![VSXzdfgvsdxf.png](https://cdn-uploads.huggingface.co/production/uploads/65bb837dbfb878f46c77de4c/nNF_6UCnmgHKjNmLaA2QA.png)
 
-**LabelPRO** is an advanced image caption and annotation generator optimized for generating detailed, structured JSON outputs. Built upon a powerful vision-language architecture with enhanced OCR and multilingual support, LabelPRO extracts high-quality captions and annotations from images for seamless integration into your applications.
+# **Caption-Pro**
+
+**Caption-Pro** is an advanced image caption and annotation generator optimized for generating detailed, structured JSON outputs. Built upon a powerful vision-language architecture with enhanced OCR and multilingual support, Caption-Pro extracts high-quality captions and annotations from images for seamless integration into your applications.
 
 #### Key Enhancements:
 
-* **Advanced Image Understanding**: Fine-tuned on millions of annotated images, LabelPRO delivers precise comprehension and interpretation of visual content.
+* **Advanced Image Understanding**: Fine-tuned on millions of annotated images, Caption-Pro delivers precise comprehension and interpretation of visual content.
 * **Optimized for JSON Output**: Produces structured JSON data containing captions and detailed annotations—perfect for integration with databases, APIs, and automation pipelines.
 * **Enhanced OCR Capabilities**: Accurately extracts textual content from images in multiple languages, including English, Chinese, Japanese, Korean, Arabic, and more.
 * **Multimodal Processing**: Seamlessly handles both image and text inputs, generating comprehensive annotations based on the provided image.
 * **Multilingual Support**: Recognizes and processes text within images across various languages.
 * **Secure and Optimized Model Weights**: Employs safetensors for efficient and secure model loading.
 
----
-
 ### How to Use
 
 ```python
 from transformers import Qwen2VLForConditionalGeneration, AutoTokenizer, AutoProcessor
 from qwen_vl_utils import process_vision_info
 
-# Load the LabelPRO model with optimized parameters
+# Load the Caption-Pro model with optimized parameters
 model = Qwen2VLForConditionalGeneration.from_pretrained(
-    "prithivMLmods/LabelPRO", torch_dtype="auto", device_map="auto"
+    "prithivMLmods/Caption-Pro", torch_dtype="auto", device_map="auto"
 )
 
 # Recommended acceleration for performance optimization:
 # model = Qwen2VLForConditionalGeneration.from_pretrained(
-#     "prithivMLmods/LabelPRO",
+#     "prithivMLmods/Caption-Pro",
 #     torch_dtype=torch.bfloat16,
 #     attn_implementation="flash_attention_2",
 #     device_map="auto",
 # )
 
-# Load the default processor for LabelPRO
-processor = AutoProcessor.from_pretrained("prithivMLmods/LabelPRO")
+# Load the default processor for Caption-Pro
+processor = AutoProcessor.from_pretrained("prithivMLmods/Caption-Pro")
 
 # Define the input messages with both an image and a text prompt
 messages = [
@@ -88,8 +87,6 @@ output_text = processor.batch_decode(
 print(output_text)
 ```
 
----
-
 ### **Key Features**
 
 1. **Annotation-Ready Training Data**  
@@ -110,6 +107,4 @@ print(output_text)
 6. **Secure and Efficient Model Weights**  
    - Uses safetensors for enhanced security and optimized model performance.
 
----
-
-**LabelPRO** streamlines the process of generating image captions and annotations, making it an ideal solution for applications that require detailed visual content analysis and structured data integration.
+**Caption-Pro** streamlines the process of generating image captions and annotations, making it an ideal solution for applications that require detailed visual content analysis and structured data integration.