aimodelshare 0.1.28__tar.gz → 0.1.30__tar.gz

{aimodelshare-0.1.28/aimodelshare.egg-info → aimodelshare-0.1.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aimodelshare
-Version: 0.1.28
+Version: 0.1.30
 Summary: Deploy locally saved machine learning models to a live rest API and web-dashboard.  Share it with the world via modelshare.org
 Home-page: https://www.modelshare.org
 Author: Michael Parrott

{aimodelshare-0.1.28 → aimodelshare-0.1.30}/aimodelshare/aimsonnx.py

@@ -579,7 +579,7 @@ def _keras_to_onnx(model, transfer_learning=None,
     output_path = os.path.join(temp_dir, 'temp.onnx')
     
     
-    model.save(temp_dir)
+    tf.saved_model.save(model, temp_dir)
 
     # # Convert the model
     try:

{aimodelshare-0.1.28 → aimodelshare-0.1.30}/aimodelshare/data_sharing/download_data.py

@@ -88,7 +88,8 @@ def pull_image(image_uri):
 	auth_head = get_auth_head(auth_url, registry, repository)
 
 	resp = requests.get('https://{}/v2/{}/manifests/{}'.format(registry, repository, tag), headers=auth_head, verify=False)
-    
+	
+	print(resp.json())
 	config = resp.json()['config']['digest']
 	config_resp = requests.get('https://{}/v2/{}/blobs/{}'.format(registry, repository, config), headers=auth_head, verify=False)
 
@@ -158,6 +159,7 @@ def pull_image(image_uri):
 
 	return docker_tar
 
+
 def extract_data_from_image(image_name, file_name, location):
     tar = tarfile.open(image_name, 'r')
     files = []
@@ -195,10 +197,11 @@ def import_quickstart_data(tutorial, section="modelplayground"):
    
     #Download Quick Start materials
     if all([tutorial == "flowers", section == "modelplayground"]):
-        quickstart_repository = "public.ecr.aws/y2e2a1d6/quickstart_materials-repository:latest"   
+        quickstart_repository = "public.ecr.aws/z5w0c9e9/quickstart_materials-repository:latest"
         existing_folder = 'flower_competition_data'
+
     if all([tutorial == "flowers", section == "competition"]):
-        quickstart_repository = "public.ecr.aws/y2e2a1d6/quickstart_flowers_competition-repository:latest"
+        quickstart_repository = "public.ecr.aws/z5w0c9e9/quickstart_flowers_competition-repository:latest"
         existing_folder = 'flower_competition_data'
     
     if all([tutorial == "mnist", section == "modelplayground"]):

{aimodelshare-0.1.28 → aimodelshare-0.1.30/aimodelshare.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aimodelshare
-Version: 0.1.28
+Version: 0.1.30
 Summary: Deploy locally saved machine learning models to a live rest API and web-dashboard.  Share it with the world via modelshare.org
 Home-page: https://www.modelshare.org
 Author: Michael Parrott

{aimodelshare-0.1.28 → aimodelshare-0.1.30}/setup.py

@@ -6,7 +6,7 @@ with open("README.md", "r") as fh:
 
 setuptools.setup(
     name='aimodelshare', #TODO:update
-    version='0.1.28',        #TODO:update
+    version='0.1.30',        #TODO:update
     author="Michael Parrott",
     author_email="mikedparrott@modelshare.org",
     description="Deploy locally saved machine learning models to a live rest API and web-dashboard.  Share it with the world via modelshare.org",
@@ -42,8 +42,8 @@ setuptools.setup(
         "protobuf==3.20.3",
         "dill==0.3.7",
         "scikeras==0.11.0"]
-      
-      
+
+
       ,
     classifiers=[
         "Programming Language :: Python :: 3",