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)')
@@ -160,8 +160,8 @@ def Custom_detect(img,mode):
 
     return Image.fromarray(im0[:,:,::-1])
 inp = gr.Image(type="pil")
-
-inp2= gr.Dropdown(choices=["Custom-Detection","Yolov7-model-detection"])
+#"Custom-Detection","Yolov7-model-detection"
+inp2= gr.Dropdown(choices=['Instance-Segmentation','Custom-Detection'])
 output = gr.Image(type="pil")
 
 examples=["Examples/Image1.jpg","Examples/Image14.jpg","Examples/Image32.jpg"]