Fixes version...

README.md

@@ -8,8 +8,7 @@ sdk_version: 6.0.2
 app_file: app.py
 pinned: false
 license: mit
-python_version: "3.10"
-duplicated_from: metal3d/labelm2
+python_version: "3.9"
 ---
 
 # 🖼️ Labelizer - AI Image Labeling Tool
@@ -27,7 +26,7 @@ An intelligent image labeling tool that uses Florence-2 vision-language model to
 ## 🚀 How to Use
 
 1. **Upload Images**: Click "📁 Upload images" to select multiple image files
-2. **Generate Labels**: 
+2. **Generate Labels**:
    - Click "✨ Generate label" below individual images
    - Or click "🏷️ Labelize all images" for batch processing
 3. **Review & Edit**: Modify any generated labels as needed
@@ -43,6 +42,7 @@ An intelligent image labeling tool that uses Florence-2 vision-language model to
 ## 📋 Supported Tasks
 
 The tool supports various captioning tasks:
+
 - `<MORE_DETAILED_CAPTION>`: Comprehensive image descriptions
 - `<DETAILED_CAPTION>`: Detailed but concise descriptions  
 - `<CAPTION>`: Basic image captions
@@ -62,4 +62,5 @@ The tool supports various captioning tasks:
 
 ---
 
-Built with ❤️ using [Gradio](https://gradio.app/) and [Florence-2](https://huggingface.co/microsoft/Florence-2-large)
+Built with ❤️ using [Gradio](https://gradio.app/) and [Florence-2](https://huggingface.co/microsoft/Florence-2-large)
+

labelizer/__init__.py

@@ -21,10 +21,10 @@ def _load_model():
 
         print(f"Loading model {MODEL_ID} on {_device} with dtype {_torch_dtype}...")
         _model = Florence2ForConditionalGeneration.from_pretrained(
-            MODEL_ID, torch_dtype=_torch_dtype, trust_remote_code=True
+            MODEL_ID, dtype=_torch_dtype, trust_remote_code=True
         ).to(_device)  # type: ignore
         _processor = Florence2Processor.from_pretrained(
-            MODEL_ID, trust_remote_code=True
+            MODEL_ID, trust_remote_code=True, use_fast=True
         )
         print("Model loaded successfully!")