Spaces:
Running
Running
File size: 780 Bytes
6a07cb2 ba5f22e 6a07cb2 ba5f22e 6a07cb2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
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, "seg.pth")
self.rec_model_path = os.path.join(
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")
@property
def get_seg_model_path(self):
return self.seg_model_path
@property
def get_rec_model_path(self):
return self.rec_model_path
@property
def get_geotr_model_path(self):
return self.geotr_model_path
@property
def get_save_path(self):
return self.save_path
|