vittoriopippi commited on
Commit
ab7810d
·
1 Parent(s): e981486

Fix import errors

Browse files
Files changed (1) hide show
  1. modeling_vatrpp.py +3 -13
modeling_vatrpp.py CHANGED
@@ -1,24 +1,14 @@
1
  from transformers import PreTrainedModel, AutoModel, AutoConfig
2
  from .configuration_vatrpp import VATrPPConfig
3
- import json
4
  import os
5
- import random
6
- import shutil
7
- from collections import defaultdict
8
- import time
9
- from datetime import timedelta
10
- from pathlib import Path
11
 
12
  import cv2
13
  import numpy as np
14
  import torch
15
 
16
- from data.dataset import FolderDataset
17
- from models.model import VATr
18
- from util.loading import load_checkpoint, load_generator
19
- from util.misc import FakeArgs
20
- from util.text import TextGenerator
21
- from util.vision import detect_text_bounds
22
  from torchvision.transforms.functional import to_pil_image
23
 
24
 
 
1
  from transformers import PreTrainedModel, AutoModel, AutoConfig
2
  from .configuration_vatrpp import VATrPPConfig
 
3
  import os
 
 
 
 
 
 
4
 
5
  import cv2
6
  import numpy as np
7
  import torch
8
 
9
+ from .data.dataset import FolderDataset
10
+ from .models.model import VATr
11
+ from .util.vision import detect_text_bounds
 
 
 
12
  from torchvision.transforms.functional import to_pil_image
13
 
14