Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import os
 
 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-e6e.pt")
 
 import argparse
 import time
@@ -21,7 +22,7 @@ from utils.torch_utils import select_device, load_classifier, time_synchronized,
 from PIL import Image
 
 parser = argparse.ArgumentParser()
-parser.add_argument('--weights', nargs='+', type=str, default='yolov7.pt', help='model.pt path(s)')
+parser.add_argument('--weights', nargs='+', type=str, default='yolov7-e6e.pt', help='model.pt path(s)')
 parser.add_argument('--source', type=str, default='Inference/', help='source')  # file/folder, 0 for webcam
 parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
 parser.add_argument('--conf-thres', type=float, default=0.25, help='object confidence threshold')
@@ -184,7 +185,7 @@ def detect(img):
 
 with torch.no_grad():
     if opt.update:  # update all models (to fix SourceChangeWarning)
-        for opt.weights in ['yolov7.pt']:
+        for opt.weights in ['yolov7-e6e.pt']:
             strip_optimizer(opt.weights)
    
 gr.Interface(detect,gr.Image(type="pil"), gr.Image(type="pil"),title="Yolov7",examples=[["horses.jpeg"]],description="demo for <a href='https://github.com/WongKinYiu/yolov7' style='text-decoration: underline' target='_blank'>WongKinYiu/yolov7</a> Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors").launch()