diff --git a/detect.py b/detect.py
index f339b4e..4f17ce6 100644
--- a/detect.py
+++ b/detect.py
@@ -149,9 +149,7 @@ def detect(opt, model, stride, save_img=False):
 
     names = ["6925303773908", "6924743915848", "6920152471616", "6920005772716", "6902227018162",
              "6920459905012", "6972194461407", "6935284412918", "6921489033706", "6904012526494",
-             "6923644272159", "6924882486100", "6956511907458", "6972636670213", "6974526401167",
-             "6914973604285", "6907476024209", "6923555210479", "6948195801276", "6976642860026",
-             "4893888001575", "6924187820067", "6921168509256"]
+             "6923644272159", "6924882486100", "6956511907458"]
     targets = []
     for target in pred[0]:
         targets.append({"Class": names[int(target[5].item())], "precision": target[4].item(), "xy1": [target[0].item(), target[1].item()],
diff --git a/ieemoo-ai-zhanting.py b/ieemoo-ai-zhanting.py
index 3aaf4f4..629ab41 100644
--- a/ieemoo-ai-zhanting.py
+++ b/ieemoo-ai-zhanting.py
@@ -40,7 +40,7 @@ app.use_reloader=False
 
 parser = argparse.ArgumentParser()
 #parser.add_argument('--weights', nargs='+', type=str, default='runs/zhanting/yolov5m_finetune/exp4/weights/best.pt', help='model.pt path(s)')
-parser.add_argument('--weights', nargs='+', type=str, default='../module/ieemoo-ai-zhanting/model/now/best_cls23_1125.pt', help='model.pt path(s)')
+parser.add_argument('--weights', nargs='+', type=str, default='../module/ieemoo-ai-zhanting/model/now/best.pt', help='model.pt path(s)')
 parser.add_argument('--source', type=str, default='../module/ieemoo-ai-zhanting/imgs/1.jpg', 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.60, help='object confidence threshold')