Update app.py

app.py

@@ -21,11 +21,11 @@ os.system('wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolo
 
 #model='best'  
 def Custom_detect(img,mode):
-    if mode=='Custom Detection':
+    if mode=='Custom-Detection':
         model='best'
-    elif mode=='Instance Segmentation':
+    if mode=='Instance-Segmentation':
         model='yolov7-mask'
-    else:
+    if mode=='Yolov7-model-detection:
         model='yolov7'
     
     parser = argparse.ArgumentParser()
@@ -160,7 +160,8 @@ def Custom_detect(img,mode):
 
     return Image.fromarray(im0[:,:,::-1])
 inp = gr.Image(type="pil")
-inp2= gr.Dropdown(default="Custom Detection",choices=["Custom Detection","Instance Segmentation","Yolov7 model detection"])
+gr.Dropdown(default="Yolo_v7_Custom_trained_By_Owais",choices=["Yolo_v7_Custom_trained_By_Owais","yolov7","yolov7-e6"])
+inp2= gr.Dropdown(default="Custom Detection",choices=["Custom-Detection","Instance-Segmentation","Yolov7-model-detection"])
 output = gr.Image(type="pil")
 
 examples=["Examples/Image1.jpg","Examples/Image14.jpg","Examples/Image32.jpg"]