BhumikaMak commited on
Commit
c17b3db
·
verified ·
1 Parent(s): d6d271e

update: added yolov8 netron support

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -48,6 +48,15 @@ def view_model(selected_models):
48
  """Generate Netron visualization for the selected models."""
49
  netron_html = ""
50
  for model in selected_models:
 
 
 
 
 
 
 
 
 
51
  if model == "yolov5":
52
  netron_html = f"""
53
  <iframe
 
48
  """Generate Netron visualization for the selected models."""
49
  netron_html = ""
50
  for model in selected_models:
51
+ if model=="yolov8":
52
+ netron_html = f"""
53
+ <iframe
54
+ src="https://netron.app/?url=https://huggingface.co/spaces/BhumikaMak/NeuralVista/resolve/main/weight_files/yolov8s.pt"
55
+ width="100%"
56
+ height="800"
57
+ frameborder="0">
58
+ </iframe>
59
+ """
60
  if model == "yolov5":
61
  netron_html = f"""
62
  <iframe