Update app.py

app.py

@@ -21,6 +21,8 @@ os.system("wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolo
 
  
 def detect_Custom(img,model):
+    if model =='Yolo_v7_Custom_trained_By_Owais':
+        model='best' # Naming Convention for yolov7
     parser = argparse.ArgumentParser()
     parser.add_argument('--weights', nargs='+', type=str, default=model+".pt", help='model.pt path(s)')
     parser.add_argument('--source', type=str, default='Inference/', help='source') 
@@ -168,8 +170,8 @@ Footer = (
     "<center> HuggingFace🤗 Model Deployed Repository <a href=\"https://huggingface.co/owaiskha9654/Yolov7_Custom_Object_Detection\">Link</a> <br></center>"
 )
 
-examples1=[["Image1.jpeg", "best"],["Image2.jpeg", "best"],["Image3.jpeg", "best",],["Image4.jpeg", "best"],["Image5.jpeg", "best"],["Image6.jpeg", "best"],["horses.jpeg", "yolov7"],["horses.jpeg", "yolov7-e6"]]
+examples1=[["Image1.jpeg", "Yolo_v7_Custom_trained_By_Owais"],["Image2.jpeg", "Yolo_v7_Custom_trained_By_Owais"],["Image3.jpeg", "Yolo_v7_Custom_trained_By_Owais",],["Image4.jpeg", "Yolo_v7_Custom_trained_By_Owais"],["Image5.jpeg", "Yolo_v7_Custom_trained_By_Owais"],["Image6.jpeg", "Yolo_v7_Custom_trained_By_Owais"],["horses.jpeg", "yolov7"],["horses.jpeg", "yolov7-e6"]]
 
 Top_Title="<center>Yolov7 Custom Trained by <a href='https://www.linkedin.com/in/owaiskhan9654/' style='text-decoration: underline' target='_blank'>Owais Ahmad</center></a>"
 
-gr.Interface(detect_Custom,[gr.Image(type="pil"),gr.Dropdown(default="best",choices=["best","yolov7","yolov7-e6"])],gr.Image(type="pil"),title=Top_Title,examples=examples1,description=Custom_description,article=Footer,cache_examples=False).launch()
+gr.Interface(detect_Custom,[gr.Image(type="pil"),gr.Dropdown(default="Yolo_v7_Custom_trained_By_Owais",choices=["best","yolov7","yolov7-e6"])],gr.Image(type="pil"),title=Top_Title,examples=examples1,description=Custom_description,article=Footer,cache_examples=False).launch()