Update app.py

app.py

@@ -17,16 +17,16 @@ from utils.torch_utils import select_device, load_classifier, time_synchronized,
 
 os.system('git clone https://github.com/WongKinYiu/yolov7')
 os.system('wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7.pt')
-os.system('wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-mask.pt')
+#os.system('wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-mask.pt')
 
 #model='best'  
 def Custom_detect(img,mode):
     if mode=='Custom-Detection':
         model='best'
-    if mode=='Instance-Segmentation':
-        model='yolov7-mask'
-    #if mode=='Yolov7-model-detection':
-       # model='yolov7'
+    #if mode=='Instance-Segmentation':
+     #   model='yolov7-mask'
+    if mode=='Yolov7-model-detection':
+        model='yolov7'
     
     parser = argparse.ArgumentParser()
     parser.add_argument('--weights', nargs='+', type=str, default=model+".pt", help='model.pt path(s)')
@@ -161,7 +161,7 @@ def Custom_detect(img,mode):
     return Image.fromarray(im0[:,:,::-1])
 inp = gr.Image(type="pil")
 #"Custom-Detection","Yolov7-model-detection"
-inp2= gr.Dropdown(choices=['Instance-Segmentation','Custom-Detection'])
+inp2= gr.Dropdown(choices=['Custom-Detection','Yolov7-model-detection'])
 output = gr.Image(type="pil")
 
 examples=[["Examples/Image1.jpg","Image1"],["Examples/Image14.jpg","Image14"],["Examples/Image32.jpg","Image32"]]