kevin-pek
/

nougat-api

vision-encoder-decoder

image-text-to-text

Inference Endpoints

Model card Files Files and versions Community

kevin-pek commited on Oct 7, 2023

Commit

835c1c9

•

1 Parent(s): 7bdd3d6

test without postprocessing

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -90,4 +90,4 @@ class EndpointHandler():
         text = self.processor.batch_decode(outputs[0], skip_special_tokens=True)[0]
         text = self.processor.post_process_generation(text, fix_markdown=False)
-        return text

         text = self.processor.batch_decode(outputs[0], skip_special_tokens=True)[0]
         text = self.processor.post_process_generation(text, fix_markdown=False)
+        return outputs