Write with transformer Arxiv model

Hi , I really loved the Write with transformers demo , especially this one(Write With Transformer) and wanted to try it out for myself , i went through the setup on sagemaker and got it running but the performance of this demo seems to be way better! Also im assuming that this (lysandre/arxiv-nlp · Hugging Face) is the model thats hosted for the demo. Would love any clarification or tips on how to achieve results similar to the demo :slight_smile: