Update app.py

app.py

@@ -293,7 +293,7 @@ def main():
     img_path = img_list[img_idx]
 
     img_name = img_path.split("/")[-1]
-    args.pretrained_path = os.path.join("/home/xli/WORKDIR/04-18/{}/cpk.pth".format(img_name.split(".")[0]))
+    args.pretrained_path = os.path.join("weights/{}/cpk.pth".format(img_name.split(".")[0]))
 
     if img_idx > -1:
         tester = Tester(args)