Visualizing model inside google Colab.

Running netron on colab and open IFrame

import IPython
import threading
import time
import os

def display_netron(path):
    os.system(f'netron {path}')
    
thread = threading.Thread(target=display_netron, args=(output_path,))
thread.start()

time.sleep(1)
display(IPython.display.IFrame(f"http://localhost:8080", width=1000, height=1000))