[Infra] introduce new env.var to set localhost url

run-docker.sh

@@ -91,6 +91,7 @@ SCRIPTPATH=$(dirname "$SCRIPT")
 : ${JUPYTER_PORT=8888}
 : ${JUPYTER_PASSWD_HASH=""}
 : ${NETRON_PORT=8081}
+: ${LOCALHOST_URL="localhost"}
 : ${PYNQ_USERNAME="xilinx"}
 : ${PYNQ_PASSWORD="xilinx"}
 : ${PYNQ_BOARD="Pynq-Z1"}
@@ -190,6 +191,7 @@ DOCKER_EXEC+="-v $FINN_HOST_BUILD_DIR:$FINN_HOST_BUILD_DIR "
 DOCKER_EXEC+="-v $FINN_SSH_KEY_DIR:/home/$DOCKER_UNAME/.ssh "
 DOCKER_EXEC+="-e FINN_BUILD_DIR=$FINN_HOST_BUILD_DIR "
 DOCKER_EXEC+="-e FINN_ROOT="/workspace/finn" "
+DOCKER_EXEC+="-e LOCALHOST_URL=$LOCALHOST_URL "
 DOCKER_EXEC+="-e VIVADO_IP_CACHE=$VIVADO_IP_CACHE "
 DOCKER_EXEC+="-e PYNQ_BOARD=$PYNQ_BOARD "
 DOCKER_EXEC+="-e PYNQ_IP=$PYNQ_IP "

src/finn/util/visualization.py

@@ -27,6 +27,7 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 import inspect
+import os
 import netron
 from IPython.display import IFrame
 
@@ -37,4 +38,5 @@ def showSrc(what):
 
 def showInNetron(model_filename):
     netron.start(model_filename, port=8081, host="0.0.0.0")
-    return IFrame(src="http://0.0.0.0:8081/", width="100%", height=400)
+    localhost_url = os.getenv("LOCALHOST_URL", default="localhost")
+    return IFrame(src="http://%s:8081/" % localhost_url, width="100%", height=400)