Spaces:

neko941
/

YOLOv5-Hololive_Waifu_Classification

Runtime error

neko941 commited on Nov 6, 2022

Commit

e54e36b

•

1 Parent(s): d5cf266

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -39,8 +39,8 @@ def check_file(file, suffix=''):
 def read_pretrain(path):
     return torch.hub.load('ultralytics/yolov5', 'custom', path=path
-default_pretrained = '2022.11.04-YOLOv5x6_1280-Hololive_Waifu_Classification.pt'
 with st.spinner('Loading the model...'):
     read_pretrain(default_pretrained)
 st.title("Hololive Waifu Classification")

 def read_pretrain(path):
     return torch.hub.load('ultralytics/yolov5', 'custom', path=path
 with st.spinner('Loading the model...'):
+    default_pretrained = '2022.11.04-YOLOv5x6_1280-Hololive_Waifu_Classification.pt'
     read_pretrain(default_pretrained)
 st.title("Hololive Waifu Classification")