Spaces:
Sleeping
Sleeping
Change line_threshold to batch_size
Browse files
app.py
CHANGED
|
@@ -55,7 +55,7 @@ def get_recognizer():
|
|
| 55 |
recognizer = TextRecognition(
|
| 56 |
model_path = TROCR_MODEL_PATH,
|
| 57 |
device = 'cuda:0',
|
| 58 |
-
|
| 59 |
)
|
| 60 |
return recognizer
|
| 61 |
except Exception as e:
|
|
|
|
| 55 |
recognizer = TextRecognition(
|
| 56 |
model_path = TROCR_MODEL_PATH,
|
| 57 |
device = 'cuda:0',
|
| 58 |
+
batch_size = 10
|
| 59 |
)
|
| 60 |
return recognizer
|
| 61 |
except Exception as e:
|