Spaces:

Uddipan107
/

ocr-reorder-space

Running

Uddipan Basu Bir commited on Apr 23

Commit

5a0deb7

1 Parent(s): 419d02f

Download checkpoint from HF hub in OcrReorderPipeline

Files changed (1) hide show

app.py CHANGED Viewed

@@ -97,10 +97,11 @@ def infer(image_path, json_file):
         # Projection → T5 decoding
         proj_feats = projection(text_feats)
-        gen_ids    = t5_model.generate(
             inputs_embeds=proj_feats,
             attention_mask=attention_mask,
-            max_length=512
         )
     # Decode to text

         # Projection → T5 decoding
         proj_feats = projection(text_feats)
+        gen_ids = t5_model.generate(
             inputs_embeds=proj_feats,
             attention_mask=attention_mask,
+            max_length=512,
+            decoder_start_token_id=t5_model.config.decoder_start_token_id
         )
     # Decode to text