Spaces:

triopood
/

testocr

Sleeping

triopood commited on Jan 25

Commit

9481512

•

1 Parent(s): 4c08581

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -228,7 +228,7 @@ training_args = Seq2SeqTrainingArguments(
 # Initialize trainer.
 trainer = Seq2SeqTrainer(
     model=model,
-    tokenizer=processor.feature_extractor,
     args=training_args,
     compute_metrics=compute_cer,
     train_dataset=train_dataset,

 # Initialize trainer.
 trainer = Seq2SeqTrainer(
     model=model,
+    tokenizer=processor.image_processor,
     args=training_args,
     compute_metrics=compute_cer,
     train_dataset=train_dataset,