dcd018 commited on
Commit
cb3d26d
Β·
1 Parent(s): d8d0c99

Initial commit.

Browse files
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- from src.inference import TableExtractionPipeline
3
 
4
  def image_classifier(inp):
5
  return {'cat': 0.3, 'dog': 0.7}
 
1
  import gradio as gr
2
+ from inference import TableExtractionPipeline
3
 
4
  def image_classifier(inp):
5
  return {'cat': 0.3, 'dog': 0.7}
src/detection_config.json β†’ detection_config.json RENAMED
File without changes
src/eval.py β†’ eval.py RENAMED
@@ -21,7 +21,7 @@ import matplotlib.patches as patches
21
  from fitz import Rect
22
  from PIL import Image
23
 
24
- sys.path.append("../detr")
25
  import util.misc as utils
26
  from datasets.coco_eval import CocoEvaluator
27
  import postprocess
 
21
  from fitz import Rect
22
  from PIL import Image
23
 
24
+ sys.path.append("./detr")
25
  import util.misc as utils
26
  from datasets.coco_eval import CocoEvaluator
27
  import postprocess
src/grits.py β†’ grits.py RENAMED
File without changes
src/inference.py β†’ inference.py RENAMED
@@ -21,7 +21,7 @@ from matplotlib.patches import Patch
21
 
22
  from main import get_model
23
  import postprocess
24
- sys.path.append("../detr")
25
  from models import build_model
26
 
27
  class MaxResize(object):
 
21
 
22
  from main import get_model
23
  import postprocess
24
+ sys.path.append("./detr")
25
  from models import build_model
26
 
27
  class MaxResize(object):
src/main.py β†’ main.py RENAMED
@@ -12,7 +12,7 @@ import numpy as np
12
  import torch
13
  from torch.utils.data import DataLoader
14
 
15
- sys.path.append("../detr")
16
  from engine import evaluate, train_one_epoch
17
  from models import build_model
18
  import util.misc as utils
 
12
  import torch
13
  from torch.utils.data import DataLoader
14
 
15
+ sys.path.append("./detr")
16
  from engine import evaluate, train_one_epoch
17
  from models import build_model
18
  import util.misc as utils
src/postprocess.py β†’ postprocess.py RENAMED
File without changes
src/structure_config.json β†’ structure_config.json RENAMED
File without changes
src/table_datasets.py β†’ table_datasets.py RENAMED
@@ -16,7 +16,7 @@ from torchvision import transforms
16
  from torchvision.transforms import functional as F
17
 
18
  # Project imports
19
- sys.path.append("detr")
20
  import datasets.transforms as R
21
 
22
 
 
16
  from torchvision.transforms import functional as F
17
 
18
  # Project imports
19
+ sys.path.append("./detr")
20
  import datasets.transforms as R
21
 
22