Rename model name...

deploy_to_hf.py

@@ -1,123 +0,0 @@
-#!/usr/bin/env python3
-"""
-Deploy docling-models-onnx to HuggingFace Hub
-"""
-
-import os
-import subprocess
-import sys
-from huggingface_hub import HfApi, Repository
-
-def run_command(cmd, description):
-    """Run a shell command and return success status"""
-    print(f"Running: {description}")
-    print(f"Command: {cmd}")
-    
-    try:
-        result = subprocess.run(cmd, shell=True, capture_output=True, text=True, cwd=os.getcwd())
-        if result.returncode == 0:
-            print(f"✓ Success: {description}")
-            if result.stdout:
-                print(f"Output: {result.stdout}")
-            return True
-        else:
-            print(f"✗ Failed: {description}")
-            print(f"Error: {result.stderr}")
-            return False
-    except Exception as e:
-        print(f"✗ Exception in {description}: {e}")
-        return False
-
-def main():
-    # Repository details
-    repo_id = "asmud/docling-models-onnx"
-    repo_url = f"https://huggingface.co/{repo_id}"
-    
-    print("=" * 60)
-    print(f"Deploying to HuggingFace Hub: {repo_id}")
-    print("=" * 60)
-    
-    # Step 1: Create repository
-    print("\n1. Creating HuggingFace repository...")
-    try:
-        api = HfApi()
-        repo_info = api.create_repo(
-            repo_id,
-            repo_type="model",
-            private=False,
-            exist_ok=True
-        )
-        print(f"✓ Repository ready: {repo_info.repo_id}")
-        print(f"  URL: https://huggingface.co/{repo_info.repo_id}")
-    except Exception as e:
-        print(f"Repository creation result: {e}")
-        # Continue anyway - repo might already exist
-    
-    # Step 2: Initialize git repository
-    print("\n2. Initializing git repository...")
-    if not os.path.exists(".git"):
-        run_command("git init", "Initialize git repository")
-    else:
-        print("✓ Git repository already initialized")
-    
-    # Step 3: Configure git user
-    print("\n3. Configuring git user...")
-    run_command('git config user.name "Asep Muhamad"', "Set git user name")
-    run_command('git config user.email "[email protected]"', "Set git user email")
-    
-    # Step 4: Add remote
-    print("\n4. Adding remote...")
-    run_command(f"git remote remove origin", "Remove existing origin (if any)")  # Ignore errors
-    run_command(f"git remote add origin {repo_url}", "Add HuggingFace origin")
-    
-    # Step 5: Setup Git LFS
-    print("\n5. Setting up Git LFS...")
-    run_command("git lfs track '*.onnx'", "Track ONNX files with LFS")
-    
-    # Step 6: Add all files
-    print("\n6. Adding files...")
-    run_command("git add .", "Add all files")
-    
-    # Step 7: Commit
-    print("\n7. Creating commit...")
-    commit_message = '''Initial release: Docling TableFormer ONNX models with JPQD quantization
-
-- Add TableFormer Accurate model (~1MB) for high-precision table structure recognition
-- Add TableFormer Fast model (~1MB) for real-time table structure recognition
-- Include comprehensive documentation and usage examples
-- JPQD quantization applied for efficient CPU inference
-- Complete Python implementation with CLI interface
-- Achieves 93.6 TEDS score on table structure recognition benchmarks
-- HuggingFace Hub compatible with proper metadata'''
-    
-    run_command(f'git commit -m "{commit_message}"', "Create initial commit")
-    
-    # Step 8: Push to HuggingFace
-    print("\n8. Pushing to HuggingFace Hub...")
-    success = run_command("git push -u origin main", "Push to HuggingFace Hub")
-    
-    if success:
-        print("\n🎉 Successfully deployed to HuggingFace Hub!")
-        print(f"Repository URL: https://huggingface.co/{repo_id}")
-    else:
-        print("\n⚠️ Push failed. Trying to pull and merge...")
-        run_command("git pull origin main --allow-unrelated-histories --no-rebase", "Pull remote changes")
-        run_command("git push origin main", "Push after merge")
-    
-    print("\n✅ Deployment completed!")
-    return True
-
-if __name__ == "__main__":
-    # Change to script directory
-    script_dir = os.path.dirname(os.path.abspath(__file__))
-    os.chdir(script_dir)
-    print(f"Working directory: {os.getcwd()}")
-    
-    try:
-        main()
-    except KeyboardInterrupt:
-        print("\n❌ Deployment cancelled by user")
-        sys.exit(1)
-    except Exception as e:
-        print(f"\n❌ Deployment failed: {e}")
-        sys.exit(1)

example.py

@@ -192,8 +192,8 @@ def main():
     
     # Model paths
     model_files = {
-        "accurate": "ds4sd_docling_models_tableformer_accurate_jpqd.onnx",
-        "fast": "ds4sd_docling_models_tableformer_fast_jpqd.onnx"
+        "accurate": "tableformer_accurate.onnx",
+        "fast": "tableformer_fast.onnx"
     }
     
     model_path = model_files[args.model]