Update Source

Author: frenchie0x4ff

.gitignore

@@ -0,0 +1,2 @@
+**/*.pyc
+**/__pycache__

easymodels/__init__.py

@@ -0,0 +1 @@
+__version__ = 1.3

easymodels/__main__.py

@@ -1,7 +1,18 @@
 from easymodels.utils import *
-
+import requests, time
+import sys
 def start():
     Menu.main()
+    
+__version__ = "1.3"
 
 if __name__ == "__main__":
+    sys.stdout.flush()
+    print('Checking For Updates...')
+    r = requests.get('https://raw.githubusercontent.com/M4cs/EasyModels/master/version.txt').text.replace('\n', '').replace('\r', '')
+    if r != __version__:
+        print('Update Available Please Run "pip install easymodules --upgrade" to update to the latest version!')
+        time.sleep(1)
+    else:
+        print('All Up To Date! Starting EasyModels CLI')
     start()

easymodels/utils/menu.py

@@ -16,7 +16,7 @@ def main():
         while True:
             print("""\
     \n\n
-    EasyModels v1.0\n
+    EasyModels v1.3\n
     \tEasyModels is a tool to quickly, and easily find pre-trained AI
     \tmodels for your project. It uses the modelzoo.co API and offers
     \tinformation about the pre-trained models in an easy to view way.
@@ -47,5 +47,5 @@ def main():
                             except:
                                 print(Categories.create_table_from_category_info(category_info))
                                 print('\nInvalid Option! Please choose an ID # from the table above!\n')
-                print('\n' * 20)
+                sys.stdout.flush()
 

setup.py

@@ -9,7 +9,7 @@ def readme():
 
 setuptools.setup(
     name="EasyModels",
-    version="1.1",
+    version="1.3",
     author="Max Bridgland",
     install_requires=[
         'requests==2.21.0',
@@ -24,7 +24,7 @@ def readme():
             'easymodels = easymodels.__main__:start'
         ]
     },
-    keywords="AI tensorflow models training pre-train pre-trained artifical intelligence keras YOLO opencv pytorch chainer",
+    keywords="AI, tensorflow, models, training, pre-train, pre-trained, artifical intelligence, keras, YOLO, opencv, pytorch, chainer, easymodels, ezmodels",
     url="https://github.com/M4cs/EasyModels",
     packages=setuptools.find_packages(),
     classifiers=(