Merge secret_model into repo

app.py

@@ -4,53 +4,20 @@ import onnx
 import onnxruntime as rt
 from torchvision import transforms as T
 from tokenizer_base import Tokenizer
-import pathlib
 from PIL import Image
-from huggingface_hub import Repository
 
 
 class DocumentParserModel:
     def __init__(
         self,
-        repo_path,
-        model_subpath,
+        model_path,
         img_size,
-        charset,
-        repo_url="stevenchang/captcha",
-        token=None,
+        charset
     ):
-        self.repo_path = pathlib.Path(repo_path).resolve()
-        self.model_path = self.repo_path / model_subpath
         self.charset = charset
         self.tokenizer_base = Tokenizer(self.charset)
-        self.initialize_repository(repo_url, token)
         self.transform = self.create_transform_pipeline(img_size)
-        self.ort_session = self.initialize_onnx_model(str(self.model_path))
-
-    def initialize_repository(self, repo_url, token):
-        if not self.model_path.exists():
-            if not self.repo_path.exists():
-                print(
-                    f"Repository does not exist. Cloning from {repo_url} into {self.repo_path}"
-                )
-                repo = Repository(
-                    local_dir=str(self.repo_path),
-                    clone_from=repo_url,
-                    use_auth_token=token if token else True,
-                )
-            else:
-                print(
-                    f"Model does not exist, but repository is already cloned. Pulling latest changes in {self.repo_path}"
-                )
-                repo = Repository(
-                    local_dir=str(self.repo_path),
-                    use_auth_token=token if token else True,
-                )
-                repo.git_pull()
-        else:
-            print(
-                f"Model {self.model_path} already exists, skipping repository update."
-            )
+        self.ort_session = self.initialize_onnx_model(str(model_path))
 
     def create_transform_pipeline(self, img_size):
         transforms = [
@@ -82,14 +49,12 @@ class DocumentParserModel:
 if __name__ == "__main__":
     import sys
 
-    repo_path = "secret_models"
-    model_subpath = "captcha.onnx"
+    model_path = "captcha.onnx"
     img_size = (32, 128)
     charset = r"0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!\"#$%&'()*+,-./:;<=>?@[\\]^_`{|}~"
 
     doc_parser = DocumentParserModel(
-        repo_path=repo_path,
-        model_subpath=model_subpath,
+        model_path=model_path,
         img_size=img_size,
         charset=charset,
     )

captcha.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2a672587ee82eb010dbef54dd0a38e99625293608ed4068c4bd20ebe467fede4
+size 95401304

requirements.txt

@@ -3,4 +3,3 @@ torchvision==0.12.0
 onnx==1.14.0
 onnxruntime==1.15.1
 Pillow==10.0.0
-huggingface-hub==0.21.4

secret_models

@@ -1 +0,0 @@
-Subproject commit 99c2c60ac74ca2b4a90613ccec0685e416dff0a8