Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ from datasets import load_dataset
|
|
2 |
from transformers import TrOCRProcessor, VisionEncoderDecoderModel, Seq2SeqTrainer, Seq2SeqTrainingArguments, default_data_collator
|
3 |
|
4 |
# Load the handwritten math dataset (1000 examples)
|
5 |
-
ds = load_dataset("Azu/Handwritten-Mathematical-Expression-Convert-LaTeX", split="train[:
|
6 |
|
7 |
# Load processor and model
|
8 |
processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-handwritten")
|
|
|
2 |
from transformers import TrOCRProcessor, VisionEncoderDecoderModel, Seq2SeqTrainer, Seq2SeqTrainingArguments, default_data_collator
|
3 |
|
4 |
# Load the handwritten math dataset (1000 examples)
|
5 |
+
ds = load_dataset("Azu/Handwritten-Mathematical-Expression-Convert-LaTeX", split="train[:100]")
|
6 |
|
7 |
# Load processor and model
|
8 |
processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-handwritten")
|