Update app.py

app.py

@@ -19,12 +19,10 @@ os.system('git clone https://github.com/WongKinYiu/yolov7')
 
 #model='best'  
 def Custom_detect(img,mode):
-    if mode=='Custom-Detection':
+    if mode=='Yolov7':
         model='best'
-    #if mode=='Instance-Segmentation':
-     #   model='yolov7-mask'
-    if mode=='Yolov7-model-detection':
-        model='yolov7'
+    if mode=='Yolov7_tiny':
+        model='tiny'
     
     parser = argparse.ArgumentParser()
     parser.add_argument('--weights', nargs='+', type=str, default=model+".pt", help='model.pt path(s)')
@@ -159,7 +157,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=['Custom-Detection','Yolov7-model-detection'])
+inp2= gr.Dropdown(choices=['Yolov7','Yolov7_tiny'])
 output = gr.Image(type="pil")
 
 examples=[["Examples/Image1.jpg","Image1"],["Examples/Image2.jpg","Image2"]]