BhumikaMak commited on
Commit
cf22491
·
verified ·
1 Parent(s): e5cf92a

debug: model name mismatch

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -48,7 +48,7 @@ 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=="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"
 
48
  """Generate Netron visualization for the selected models."""
49
  netron_html = ""
50
  for model in selected_models:
51
+ if model=="yolov8s":
52
  netron_html = f"""
53
  <iframe
54
  src="https://netron.app/?url=https://huggingface.co/spaces/BhumikaMak/NeuralVista/resolve/main/weight_files/yolov8s.pt"