Add .mlmodel extension and setup

electron-builder.yml

@@ -25,6 +25,9 @@ fileAssociations:
     - keras
     - h5
     - json
+  - name: "CoreML Model"
+    ext:
+    - coreml
 publish:
   - provider: github
     releaseType: release

setup.py

@@ -75,7 +75,7 @@ setuptools.setup(
             'tf-model.js', 'tf.js', 'tf-operator.pb',
             'tflite-model.js', 'tflite.js', 'tflite-operator.json',
             'keras-model.js', 'keras-operator.json', 'hdf5.js',
-            'coreml-model.js', 'coreml-operator.json',
+            'coreml-model.js', 'coreml-operator.json', 'coreml.js',
             'view-browser.html', 'view-browser.js',
             'view.js', 'view.css', 'view-render.css', 'view-render.js', 'view-template.js'
         ]

src/app.js

@@ -111,7 +111,8 @@ class Application {
                 { name: 'TensorFlow Saved Model', extensions: [ 'saved_model.pb' ] },
                 { name: 'TensorFlow Graph', extensions: [ 'pb', 'meta' ] },
                 { name: 'TensorFlow Lite Model', extensions: [ 'tflite' ] },
-                { name: 'Keras Model', extension: [ 'json', 'keras', 'h5' ] }
+                { name: 'Keras Model', extension: [ 'json', 'keras', 'h5' ] },
+                { name: 'CoreML Model', extension: [ 'mlmodel' ] },
             ]
         };
         electron.dialog.showOpenDialog(showOpenDialogOptions, (selectedFiles) => {

src/view-browser.html

@@ -38,6 +38,7 @@
 <script type='text/javascript' src='tf-model.js'></script>
 <script type='text/javascript' src='tflite-model.js'></script>
 <script type='text/javascript' src='keras-model.js'></script>
+<script type='text/javascript' src='coreml-model.js'></script>
 <script type='text/javascript' src='view-template.js'></script>
 <script type='text/javascript' src='view-browser.js'></script>
 <script type='text/javascript' src='view-render.js'></script>