Spaces:

Sa-m
/

Vehicles-Detection-Custom-YoloV7

Runtime error

Sa-m commited on Aug 22, 2022

Commit

2f25530

1 Parent(s): b54919c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ def Custom_detect(img,mode):
         model='best'
     if mode=='Instance-Segmentation':
         model='yolov7-mask'
-    if mode=='Yolov7-model-detection:
         model='yolov7'
     parser = argparse.ArgumentParser()

         model='best'
     if mode=='Instance-Segmentation':
         model='yolov7-mask'
+    if mode=='Yolov7-model-detection':
         model='yolov7'
     parser = argparse.ArgumentParser()