Feat/python package (#23)

* feat: python package * chore: specify branch name
elazarg · Jul 17, 2024 · 6dc2514 · 6dc2514
1 parent df2f202
commit 6dc2514
Show file tree

Hide file tree

Showing 7 changed files with 29 additions and 7 deletions.
diff --git a/.gitignore b/.gitignore
@@ -8,4 +8,6 @@ group*-shard*
 models
 final_model
 wandb
-cert
+cert
+venv/
+nakdimon.egg*
diff --git a/examples/usage.py b/examples/usage.py
@@ -0,0 +1,7 @@
+# pip install git+https://github.com/thewh1teagle/nakdimon@feat/python-package
+# wget https://github.com/elazarg/nakdimon/raw/master/models/Nakdimon.h5
+
+import nakdimon
+
+result = nakdimon.predict("Nakdimon.h5", "שלום עולם!")
+print(result)
diff --git a/nakdimon/__init__.py b/nakdimon/__init__.py
@@ -0,0 +1,3 @@
+from .predict import predict
+
+__all__ = ['predict']
diff --git a/nakdimon/dataset.py b/nakdimon/dataset.py
@@ -3,8 +3,8 @@
 import random
 import numpy as np
 
-import hebrew
-import utils
+from nakdimon import hebrew, utils
+
 
 
 class CharacterTable:

diff --git a/nakdimon/hebrew.py b/nakdimon/hebrew.py
@@ -6,7 +6,7 @@
 from functools import lru_cache
 import re
 
-import utils
+from nakdimon import utils
 
 
 # "rafe" denotes a letter to which it would have been valid to add a diacritic of some category

diff --git a/nakdimon/predict.py b/nakdimon/predict.py
@@ -3,9 +3,7 @@
 
 import tensorflow as tf
 
-import utils
-import dataset
-import hebrew
+from nakdimon import utils, dataset, hebrew
 
 
 if tf.config.set_visible_devices([], 'GPU'):

diff --git a/setup.py b/setup.py
@@ -0,0 +1,12 @@
+from setuptools import setup, find_packages
+
+setup(
+    name='nakdimon',
+    version='0.1.0',
+    packages=find_packages(),
+    install_requires=[
+        "numpy==1.24.1",
+        "tensorflow==2.13.0",
+        "wandb==0.17.0"
+    ],
+)