Added file format restriction for upload

* Allows only '.prototxt' file for Caffe * Allows only '.pbtxt' file for Tensorflow
Cloud-CV · Jun 24, 2017 · a51edde · a51edde
1 parent 7d00920
commit a51edde
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/ide/static/bundle/bundle.js b/ide/static/bundle/bundle.js
@@ -43719,7 +43719,7 @@
 	                          { htmlFor: "inputFilecaffe" },
 	                          "caffe"
 	                        ),
-	                        _react2.default.createElement("input", { id: "inputFilecaffe", type: "file", onChange: function onChange() {
+	                        _react2.default.createElement("input", { id: "inputFilecaffe", type: "file", accept: ".prototxt", onChange: function onChange() {
 	                            return _this2.props.importNet('caffe');
 	                          } })
 	                      )
@@ -43735,7 +43735,7 @@
 	                          { htmlFor: "inputFiletensorflow" },
 	                          "tensorflow"
 	                        ),
-	                        _react2.default.createElement("input", { id: "inputFiletensorflow", type: "file", onChange: function onChange() {
+	                        _react2.default.createElement("input", { id: "inputFiletensorflow", type: "file", accept: ".pbtxt", onChange: function onChange() {
 	                            return _this2.props.importNet('tensorflow');
 	                          } })
 	                      )

diff --git a/ide/static/js/topBar.js b/ide/static/js/topBar.js
@@ -31,13 +31,13 @@ class TopBar extends React.Component {
                     <li>
                         <a>
                         <label htmlFor="inputFilecaffe">caffe</label>
-                        <input id="inputFilecaffe" type="file" onChange={() => this.props.importNet('caffe')}/>
+                        <input id="inputFilecaffe" type="file" accept=".prototxt" onChange={() => this.props.importNet('caffe')}/>
                         </a>
                     </li>
                     <li>
                         <a>
                         <label htmlFor="inputFiletensorflow">tensorflow</label>
-                        <input id="inputFiletensorflow" type="file" onChange={() => this.props.importNet('tensorflow')}/>
+                        <input id="inputFiletensorflow" type="file" accept=".pbtxt" onChange={() => this.props.importNet('tensorflow')}/>
                         </a>
                     </li>
                     <li><a href="#" onClick={() => this.props.importNet('url')}>url</a></li>