Update run.py
Browse files
run.py
CHANGED
@@ -9,6 +9,7 @@ import time
|
|
9 |
processor = TrOCRProcessor.from_pretrained("arcma/decap")
|
10 |
model = VisionEncoderDecoderModel.from_pretrained("arcma/decap")
|
11 |
model.eval()
|
|
|
12 |
|
13 |
def check(x):
|
14 |
if len(x) < 6:
|
|
|
9 |
processor = TrOCRProcessor.from_pretrained("arcma/decap")
|
10 |
model = VisionEncoderDecoderModel.from_pretrained("arcma/decap")
|
11 |
model.eval()
|
12 |
+
torch.compile(model)
|
13 |
|
14 |
def check(x):
|
15 |
if len(x) < 6:
|