add models

config.py

@@ -4,9 +4,9 @@ import os
 class Config:
     def __init__(self):
         self.current_dir = os.path.dirname(os.path.abspath(__file__))
-        self.seg_model_path = os.path.join(self.current_dir, "pretrained", "seg.pth")
+        self.seg_model_path = os.path.join(self.current_dir, "seg.pth")
         self.rec_model_path = os.path.join(
-            self.current_dir, "pretrained", "DocScanner-L.pth"
+            self.current_dir, "DocScanner-L.pth"
         )
         self.geotr_model_path = os.path.join(self.current_dir, "pretrained", "model.pt")
         self.save_path = os.path.join(self.current_dir, "output")